Type of Crime Classification AI hackathon

Algorithm | Tabular | Classification | Social | Macro_F1_Score

Prize : 인증서
589 Users Completed

Practice

Overview Data Code (share) Talk Leaderboard

Submission

[Private 2위] CatBoost + optuna

9mklee

2023.05.30 15:17 2,609 Views language

- 데이터 전처리는 레이블 인코딩만 진행했고, 타겟 변수 불균형 문제가 있어 클래스별 가중치를 적용해 모델 학습을 진행했습니다.
- 모델은 LightGBM과 CatBoost를 사용해봤는데 CatBoost가 조금 더 높게 나와서 최종 모델로 선택했습니다.
- 하이퍼 파라미터 튜닝은 optuna를 사용했고, 코랩 환경에서 진행했습니다.

감사합니다.

Code