/GDSRec_rank

Primary LanguagePython

A PyTorch implementation of GDSRec for Ranking Task

  1. Preprocess dataset.
python preprocess.py
  1. Run main.py file to train the model. You can configure some training parameters through the command line.
python main.py --test=False
  1. Run main.py file to test the model.
python main.py --test=True

The hyper-parameter "cut_off" in ratingt_data.py corresponds to the parameter "F" in the paper. It controls the threshold of rating for splitting the data.