- 사용자의 책 평점 데이터를 바탕으로 사용자가 어떤 책을 더 선호할지 예측하는 태스크입니다.
- 해당 경진대회는 이러한 소비자들의 책 구매 결정에 대한 도움을 주기 위한 개인화된 상품 추천 대회입니다.
- 리더보드는 평점 예측에서 자주 사용되는 지표 중 하나인 RMSE (Root Mean Square Error)를 사용합니다.
- 김동현
- 데이터 전처리, 기본 제공 모델 실험, sklearn의 여러 머신러닝 모델 적용
- 임우열
- 데이터 전처리, 후처리, GBM 계열 Model, SVD
- 장형규
- 데이터 전처리, FM, FFM 모델 실험, Catboost Regression 적용, 시각화
- 채민지
- 데이터 전처리, FM, CNN_FM, SVD 모델 실험
- 황선우
- 데이터 전처리, CNN FFM 모델 설계 및 실험
- ai stage server : V100 GPU x 4
- python==3.8.5
- torch==1.7.1
- CUDA==11.0