Skip to content

MEICRS/GDSRec_rank

Repository files navigation

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.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages