자율주행 센서의 안테나 성능 예측 AI 경진대회

알고리즘 | 정형 | 회귀 | 자율주행 | NRMSE

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

 

[Private기준 28등, Public기준 36등] Attention을 기반으로 Ensemble 진행

공동작성자

stroke
2022.10.02 15:00 1,931 조회 language

안녕하세요!
저희는 Private기준 28등, Public기준 36등을 기록하였습니다.
도메인 지식의 부족으로 feature engineering을 다각적으로 진행하지는 못하였지만,
모델 앙상블을 할 때 attention을 기반으로 해보았는데, 아이디어를 공유할겸 코드를 공유하게 되었습니다!
최종모델은 lgbm, xgboost, random forest를 optuna 돌려 최적의 파라미터를 추출하고 이 셋 모델을 앙상블하는 방법을 택했지만,
이 코드는 파생변수를 다각적으로 생성하여 데이터셋을 추가로 다섯개로 만들고,
이에 대한 xgboost prediction 값을 attention score를 기반으로 ensemble한 것입니다!

읽어주셔서 감사합니다!

코드