This code is released for the papers(under review):
DiffRSG: Rules-based Skip-GCN in Neural Latent Information Diffusion Network for Social Recommendation
The article is in the period of review. Please do not use it for other purposes
- Python==3.6
- Tensorflow-gpu==1.12.0
Download datasets (Yelp, Flickr) from this link , and just put the downloaded folder 'data' in the sub-directory named DiffRSG of your local clone repository.
- GPU: RTX5000 (16GB)
- GPU: RTX3090 (24GB)
- Run DiffRSG: cd DiffRSG directory execute the command
python entry.py --data_name=<data_name> --model_name=DiffRSG --gpu=<gpu id>
- If you have any available gpu device, you can specify the gpu id, or you can just ignore the gpu id. Following are the command examples:
python entry.py --data_name=yelp --model_name=DiffRSG --gpu=1
python entry.py --data_name=flickr --model_name=DiffRSG --gpu=2