A PyTorch implementation of the GraphRec model in Graph Neural Networks for Social Recommendation (Fan, Wenqi, et al. "Graph Neural Networks for Social Recommendation." The World Wide Web Conference. ACM, 2019).
- Install required packages from requirements.txt file.
pip install -r requirements.txt- 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- Run main.py file to train the model. You can configure some training parameters through the command line.
python main.py- Run main.py file to test the model.
python main.py --test
![@dependabot[bot]](https://avatars.githubusercontent.com/in/29110?s=64&v=4){kind=small}