잡케어 추천 알고리즘 경진대회

알고리즘 | 정형 | 분류 | 추천 | Macro f1 score

  • moneyIcon 상금 : 총 1,000만원
  • 1,629명 마감

 

Model Selection + LightGBM + Optuna (Score: 0.694)

2022.01.28 16:11 3,754 조회 language

Colab GPU 런타임에서 사용했으며, LightGBM / XGBoost / CatBoost를 모두 GPU 환경에서 Train했습니다.
Optuna라는 모듈과 Cross validation을 활용해 hyper-parameter optimization을 수행했으며, model selection 과정 역시 포함했습니다.
또한 LASSO를 통해 계수가 0인 변수를 제거하는 Feature selection 역시 진행을 했는데, 생각만큼 성능이 그렇게 올라가진 않았네요^^;;
부족한 부분이 있다면 조언해주시면 감사하겠습니다~

코드