김찬호 | 배성수 | 이지훈 | 정소빈 | 조원삼 |
협업 관리, 인퍼런스 구현, EASE, 앙상블 | 모델 탐색, 데이터 전처리, 모델 베이스라인 개발 및 실험, 앙상블 | 모델 탐색 및 실험, Nue-MF Pytorch Project 개발 | 모델 탐색 및 실험 | EDA, DeepFM, Bert4rec |
사용자의 영화 시청 이력 데이터를 바탕으로 사용자가 다음에 시청할 영화 및 좋아할 영화를 예측
- 사용자의 영화 시청 이력 데이터 5,154,471 개
- 영화 아이템 메타 정보 데이터 6,807 개
- AutoEncoder 계열: Multi-VAE, Multi-DAE, MSE-DAE, EASE, ALS
- 시퀀셜 모델: Sasrec, S3rec, Bert4rec
- Top-K의 결과를 기반으로 Hard Voting 방법 사용
- 특성이 다른 시퀀셜과 AE 기반 모델을 사용하여 앙상블
리더보드 | Recall@10 | 순위 |
---|---|---|
public | 0.1646 | 6위 |
private | 0.1634 | 최종 5위 |