We sincerely thank for the robust support provided by the SELFRec framework for this project.
SELFRec is a Python framework for self-supervised recommendation (SSR) which integrates commonly used datasets and metrics, and implements many state-of-the-art SSR models. SELFRec has a lightweight architecture and provides user-friendly interfaces. It can facilitate model implementation and evaluation.
Founder and principal contributor: @Coder-Yu @xiaxin1998
Supported by: @AIhongzhi (A/Prof. Hongzhi Yin, UQ)
To learn more about the Self Rec framework, please visit https://github.com/Coder-Yu/SELFRec/.
numba==0.53.1
numpy==1.20.3
scipy==1.6.2
torch>=1.7.0
- Configure the xx.conf file in the directory named conf. (xx is the name of the model you want to run)
- Run main.py and choose the model you want to run.
| Model | Paper | Type | Code |
|---|---|---|---|
| DSVC | Yang et al. Dual Social View Enhanced Contrastive Learning for Social Recommendation, TCSS'24. | Graph + CL | PyTorch |
| Datasets | Yelp2018 | Douban-book | FilmTrust |
|---|---|---|---|
| # User | |||
| # Item | |||
| # Interaction | |||
| # Relation | |||
| U-I Density | |||
| U-U Density |
If you find this repo helpful to your research, please cite our paper.
@article{yang2024dual,
title={Dual Social View Enhanced Contrastive Learning for Social Recommendation},
author={Yang, Shixiao and Qin, Zhida and Du, Enjun and Zhou, Pengzhan and Huang, Tianyu},
journal={IEEE Transactions on Computational Social Systems},
year={2024},
publisher={IEEE}
}
