Skip to content

Latest commit

 

History

History
26 lines (19 loc) · 627 Bytes

README.md

File metadata and controls

26 lines (19 loc) · 627 Bytes
Raw

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