음악 장르 분류 AI 해커톤

롤케잌 | Private 4위 (0.65815) | 부스팅 모델 | Over-Sampling

2023.02.01 15:48 885 Views language

안녕하세요, private 4위 코드 공유합니다
부스팅 모델로 xgboost, lightgbm, catboost를 사용했습니다
샘플 수가 가장 적은 클래스만 오버샘플링을 통해 그 다음으로 적은 클래스의 샘플 수로 맞춰주었습니다 (다른 클래스에도 오버샘플링을 적용한 경우엔 성능이 감소했습니다)

아래의 코드는 google colab에서 작성했습니다, 감사합니다  

Code