sungsubae / boostcamp-movierecommendation-recsys-09

[Boostcamp AI Tech] Movie Recommendation project

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

스크린샷 2023-01-06 오전 9 15 45

9️⃣ boostcamp AI Tech 4th - RecSys

👪 Members

김찬호 배성수 이지훈 정소빈 조원삼
협업 관리, 인퍼런스 구현, EASE, 앙상블 모델 탐색, 데이터 전처리, 모델 베이스라인 개발 및 실험, 앙상블 모델 탐색 및 실험, Nue-MF Pytorch Project 개발 모델 탐색 및 실험 EDA, DeepFM, Bert4rec

🎬 Movie Recommendation

사용자의 영화 시청 이력 데이터를 바탕으로 사용자가 다음에 시청할 영화 및 좋아할 영화를 예측


📄 Data

  • 사용자의 영화 시청 이력 데이터 5,154,471 개
  • 영화 아이템 메타 정보 데이터 6,807 개

💻 Repository Summary

코드 구조도 drawio


🗃 Project Process

🤖 Model

  • AutoEncoder 계열: Multi-VAE, Multi-DAE, MSE-DAE, EASE, ALS
  • 시퀀셜 모델: Sasrec, S3rec, Bert4rec

📈 Ensemble

  • Top-K의 결과를 기반으로 Hard Voting 방법 사용
  • 특성이 다른 시퀀셜과 AE 기반 모델을 사용하여 앙상블

🏅 Result : Public 6th > Privite 5th

리더보드 Recall@10 순위
public 0.1646 6위
private 0.1634 최종 5위

스크린샷 2023-01-08 오후 11 45 27


About

[Boostcamp AI Tech] Movie Recommendation project


Languages

Language:Jupyter Notebook 92.1%Language:Python 7.9%