R-GSN:异构图的基于关系的图相似网络
这是纸的代码
1.环境
Ubuntu16.04 + NVIDIA-1080TI Anaconda + Python 3.7
numpy==1.19.4
scipy==1.5.4
ogb==1.2.3
texttable==1.6.3
torch==1.6.0+cu101
torchvision==0.7.0+cu101
torch-cluster==1.5.8
torch-geometric==1.6.1
torch-scatter==2.0.5
torch-sparse==0.6.8
torch-spline-conv==1.2.0
2.用法
运行R-GSN
python rgsn.py --device $Device --conv_name " rgsn " \
--test_batch_size 64 \
--Norm4 True \