범죄 유형 분류 AI 해커톤

알고리즘 | 정형 | 분류 | 사회 | Macro F1 Score

상금 : 인증서
528명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

[Private 2위] CatBoost + optuna

9mklee

2023.05.30 15:17 1,105 조회 language

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

감사합니다.

코드