Skip to content

ardkyer/NaverBoostcamp_BookRating_Prediction_AI_Project

 
 

Repository files navigation

📚 Book Recommendation

책과 관련된 정보와 소비자의 정보, 그리고 소비자가 실제로 부여한 평점을 활용하여 사용자가 주어진 책에 대해 얼 마나 평점을 부여할지에 대해 예측합니다. 해당 경진대회는 소비자들의 책 구매 결정에 대한 도움을 주기 위한 개인화된 상품 추천 대회입니다.

자세한 프로젝트 내용은 Wrap-up Report발표PPT를 통해 확인해주세요.

💡Team

팀 구성

강현구 서동준 이도걸 이수미 최윤혜

역할 분담

이름 역할
강현구 EDA, DeepFM, ResNet_DeepFM
서동준 TextFM, TextDeepFM, LGBM
이도걸 WDN, DCN, NCF, Boosting
이수미 Add features, FM, FFM
최윤혜 Data Preprocessing, ImageFM, Image DeepFM, CatBoost

📂 Architecture

.
📦 Project Root
┣ 📜 main.py
┣ 📂 src
┃ ┣ 📂 data
┃ ┃ ┣ 📊 basic_data.py
┃ ┃ ┣ 📊 context_data.py
┃ ┃ ┣ 🖼️ image_data.py
┃ ┃ ┗ 📝 text_data.py
┃ ┣ 📂 ensembles
┃ ┃ ┗ 🤝 ensembles.py
┃ ┣ 📂 loss
┃ ┃ ┗ 📉 loss.py
┃ ┣ 📂 models
┃ ┃ ┣ 🛠️ _helpers.py
┃ ┃ ┣ 🔮 DCN.py
┃ ┃ ┣ 🔮 DeepFM.py
┃ ┃ ┣ 🔮 FFM.py
┃ ┃ ┣ 🖼️ FM_Image.py
┃ ┃ ┣ 📝 FM_Text.py
┃ ┃ ┣ 🔮 FM.py
┃ ┃ ┣ 🔮 NCF.py
┃ ┃ ┗ 🔮 WDN.py
┃ ┣ 📂 train
┃ ┃ ┗ 🎯 trainer.py
┃ ┗ 🛠️ utils.py
┣ 📜 ensemble.py
┣ 📜 run_baseline.sh
┣ 📓 eda
┃ ┗ 📊 eda.ipynb
┗ 📂 config
┣ ⚙️ config_baseline
┗ ⚙️ sweep_example

⚒️ Development Environment

  • 서버 스펙 : AI Stage GPU (Tesla V100)
  • 협업 툴 : Github / Zoom / Slack / Notion / Google Drive
  • 기술 스택 : Python / Scikit-Learn / Scikit-Optimize / Pandas / Numpy / PyTorch

About

Book Recommendation Competition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 65.1%
  • Jupyter Notebook 34.9%