/GDSRec

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

A PyTorch implementation of GDSRec

  1. Install required packages from requirements.txt file.
pip install -r requirements.txt
  1. Preprocess dataset. Two pkl files named dataset and list should be generated in the respective folders of the dataset.
python preprocess.py --dataset Ciao
python preprocess.py --dataset Epinions
  1. Run main.py file to train the model. You can configure some training parameters through the command line.
python main.py
  1. Run test.py file to test the model.
python test.py

For the ranking task, please see https://github.com/MEICRS/GDSRec_rank