Source code (PyTorch) and dataset of the paper "MGDCF: Distance Learning via Markov Graph Diffusion for Neural Collaborative Filtering", which is accepted by IEEE Transactions on Knowledge and Data Engineering (TKDE).
- PyTorch Implementation: Torch-MGDCF
- TensorFlow Implementation: TensorFlow-MGDCF
- Paper Access:
- IEEE Xplore: https://ieeexplore.ieee.org/document/10384729
- ArXiv: https://arxiv.org/abs/2204.02338
I have tested this environment with the following requirements, but please note that you are not restricted to these specific versions and can choose other versions as appropriate, since my code does not utilize many advanced operations:
- Linux
- Python 3.7
- torch=1.12.1+cu113
- numpy >= 1.17.4
- tqdm
- dgl=1.1.1+cu113
- requests=2.28.1
- scikit-learn=1.0.2
- faiss-cpu=1.7.4
sh run_mgdcf_yelp.sh
sh run_mgdcf_gowalla.sh
sh run_mgdcf_amazon-book.sh@ARTICLE{10384729,
author={Jun Hu and Bryan Hooi and Shengsheng Qian and Quan Fang and Changsheng Xu},
journal={IEEE Transactions on Knowledge and Data Engineering},
title={MGDCF: Distance Learning via Markov Graph Diffusion for Neural Collaborative Filtering},
year={2024},
volume={},
number={},
pages={1-16},
doi={10.1109/TKDE.2023.3348537}
}
License: GPLv3
Copyright (c) 2023-2024 Xtra Computing Group, NUS, Singapore.