SR-HGNN

ICDM-2020 《Global Context Enhanced Social Recommendation with Hierarchical Graph Neural Networks》

Environments

• python 3.8
• pytorch-1.6
• DGL 0.5.3 (https://github.com/dmlc/dgl)

Example to run the codes

dataprocess:

python dataProcess.py --dataset [CiaoDVD,Epinions,Douban] --rate [0.8,0.6]

train SR-HGNN model:

python main.py --dataset CiaoDVD --rate 0.8 --batch 128 --lr 0.001 --layer [16,16] --r 0.001 --r2 0.2 --r3 0.01 --dgi_hide_dim 500 --dgi_lr 0.001 --lam_t 0.05 --lam_r 0.1