AI프렌즈 시즌 공공 데이터 활용 온도 추정 AI 경진대회

알고리즘 | 정형 | 회귀 | 기상 | MSE

  • moneyIcon 상금 : 총 250만원
  • 1,608명 마감

 

LGBM에 custom metric 적용하는 코드

2020.03.06 21:51 10,214 조회 language

대회 평가 metric은 기본적으로는 mse지만 차이가 1이하인 값들은 무시합니다. 

DACON측에서 metric을 공개했지만 이를 lgbm에 바로 적용할 수 없어 적합하게 코드를 짜보았습니다.

현재까지 결과로는 오버핏 되는 경향이 있는데 추후에 도움될까싶어 공유합니다.
** 정정합니다. validation set에서 점수가 더 높아서 이렇게 생각했는데 제출한 score에서 미세하게 높아졌습니다.


XGboost에도 동일하게 적용된다고 합니다.



참고 주소: https://www.kaggle.com/c/bigcontest2019/discussion/103763

코드
로그인이 필요합니다
0 / 1000
SDSTony
2020.03.06 22:12

안녕하세요 코드 잘 보았습니다. 
score = np.sqrt(mean_squared_error(y_true, y_pred)) 부분에서 np.sqrt가 포함이 되면 루트를 씌어주니까 RMSE가 되는 것으로 알고 있습니다. 
큰 문제는 없겠지만, 실질적인 대회 평가 지표는 np.sqrt()를 제외 했을 때 나오는 MSE 값으로 산출 되는 것 같습니다

SDSTony
2020.03.06 22:17

직관적으로 y_true랑 y_pred랑 차이가 얼마나 나오는지 보기 위해서 일부러 np.sqrt() 씌어 두신 거라면 죄송합니다.. ㅎㅎ

호구너
2020.03.06 22:47

앗 제 코드에서는 수정했는데 이 코드에서는 수정이 안되었네요!
감사합니다! 

SDSTony
2020.03.06 23:16

네~ 확인 감사합니다~

최정명
2020.03.07 00:29

커스텀 loss를 적용 어떻게 하나 찾아보려고 했는데 마침 올려주셨네요 !!! 감사합니다.