스마트 공장 제품 품질 상태 분류 AI 온라인 해커톤

채용 | 알고리즘 | 정형 | 스마트 공장 | LG Aimers | 분류 | Macro F1 Score

  • moneyIcon 상금 : 본선 진출
  • 1,444명 마감

 

[Private 2위] catboost + optuna

공동작성자

stroke
2023.04.01 18:30 4,606 조회 language

Time 변수를 활용해서 5~6개의 파생변수를 만들었고, 마지막은 CatboostRegressor와 CatboostClassifier를 활용하여 각각 예측하고,
비교한 후에 제품의 Spec에서 많이 벗어난 레이블을 변경하여 제출했습니다.
Regressor는 Optuna를 적용했고, Classifier 같은 경우에는 Optuna를 적용했을 때 시간이 오래 걸려서 단순 Classifier로 진행했습니다.
그리고 Overfitting을 막고자 Stratified K-fold(K=10)를 진행했습니다.

코드