Code for paper "ConsisRec: Enhancing GNN for Social Recommendation via Consistent Neighbor Aggregation"
bash train.sh
- python 3.8
- torch 1.7.1
- numpy 1.20.0
- sklearn
- Code for some traditional and social recommendation methods
- Code for GraphRec
- Paper summary for social recommendation
If you use this code, please cite:
paper link: https://arxiv.org/pdf/2105.02254
@inproceedings{yang2021consisrec,
title={ConsisRec: Enhancing GNN for Social Recommendation viaConsistent Neighbor Aggregation},
author={Yang, Liangwei and Liu, Zhiwei and Dou, Yingtong and Ma, Jing and Philip S. Yu},
journal={Proceedings of the 44th international ACM SIGIR conference on Research and development in information retrieval},
year={2021},
publisher={ACM}
}
paper link: https://arxiv.org/pdf/1902.07243.pdf
@inproceedings{fan2019graph,
title={Graph neural networks for social recommendation},
author={Fan, Wenqi and Ma, Yao and Li, Qing and He, Yuan and Zhao, Eric and Tang, Jiliang and Yin, Dawei},
booktitle={The World Wide Web Conference},
pages={417--426},
year={2019}
}