/IGMC_CX

Primary LanguageJupyter Notebook

Improving Graph-Based Movie Recommender System Using Cinematic Experience

Dependencies

  • PyTorch 1.2+
  • DGL 0.5 (nightly version)

Data

Movie data

Training set for BERT

How to run

  • explicit_train_rotten.ipynb

Results

Dataset Our code
best of epochs
Rotten Tomatoes 0.8004
Amazon Movie 0.9621