GitHub - boostcampaitech4recsys1/level1_bookratingprediction_recsys-level1-recsys-02: level1_bookratingprediction_recsys-level1-recsys-02 created by GitHub Classroom

• Public LB: 11위 Private LB: 12위 (14팀 참여)
• 평가 Metric: RMSE

Wrap-Up Report

1. 프로젝트 개요

1.1 프로젝트 주제

본 프로젝트는 사용자의 책 평점 데이터를 바탕으로 사용자가 어떤 책을 더 선호할지 예측하는 모델을 구축하고자 한다.

책에 대한 메타 데이터인 books, 고객에 대한 메타 데이터인 users, 고객이 책에 남긴 평점 ratings 의 데이터 셋을 활용해, 최종적으로 1과 10 사이 평점을 예측하는 것을 목적으로 한다.

(데이터셋 출처: https://www.kaggle.com/datasets/ruchi798/bookcrossing-dataset)

1.2 프로젝트 개요

• 제공된 Base line 코드 및 decision tree와 같은 여러 모델들을 활용해 다양한 실험을 진행했다.
• 실험을 통해 NCF 모델을 기반으로 하여 context를 활용한 NCF 모델을 최종적으로 활용했다.

1.3 활용 장비 및 개요

• GPU: V100 5대
• 운영체제: Ubuntu 18.04.5 LTS