SW중심대학 공동 AI 경진대회 ❮예선❯

SW중심대학 | 예선 | 심리성향 | AUC

  • moneyIcon 상금 : 총 0만원
  • 772명 마감

 

[인공무능, Private 3위] RandomForest + ExtraTree + XGBoost

공동작성자

stroke
2022.08.29 18:27 1,215 조회 language

데이터 분석을 통한 데이터 전처리 후 RandomForest, ExtraTree, XGBoost를 Stacking으로 결합해 예측했습니다.
데이터 전처리는 NN모델 개발을 위해서 0부터 1사이의 float값으로 처리를 했으나 해당 데이터에 대해 NN모델의 성능이 좋지 않아 전처리만 가져와 사용하였습니다.
모델 파라미터 튜닝은 GridSearchCV를 사용해서 수동으로 처리했으며, LightGBM 튜닝이 마무리가 안되어 LightGBM 까지는 결합하지 못하였습니다.
해당 코드는 Colab을 사용하여 개발되었습니다.

코드