감귤 착과량 예측 AI 경진대회

알고리즘 | 정형 | 회귀 | NMAE

  • moneyIcon 상금 : 500 만원
  • 902명 마감

 

[Private 20위] Simple is the Best(XGB, LGB)

2022.12.14 17:54 1,271 조회 language

기간이 짧았던 대회인 만큼 최대한 간단하게 문제를 해결해보려고 하였다.
Parameter for XGB: learning_rate = 0.05, max_depth = 5, n_estimators = 100
Parameter for LGB: learning_rate = 0.01, max_depth = 10, n_estimators = 500, num_leaves = 31

상관관계 점수 0.5이상, -0.5이하인 feature들만 학습에 사용했다.

PDF
코드