PyTorch中的RAHGCN
1.概述
该存储库是PyTorch中通过强化学习(RAHGCN)实现的自适应双曲图卷积神经网络。 下游任务包括:
链接预测( lp )
节点分类( nc )
2.设定
2.1下载代码
首先从Github下载源代码。
git clone git@github.com:fuxingcheng/RHGNN.git"
cd rahgcn
2.2启动虚拟环境
我们建议在虚拟环境中设置我们的项目。 您可以选择conda或virtualenv来创建和管理虚拟环境。 如果您尚未安装conda,请按照的说明进行安装。 如果尚未安装virtualenv,则只需运行pip3 install virtualenv 。
对于conda :
conda env create -f environment.yml python=3.6
source activate ra
2022-10-26 21:08:44
6.5MB
Python
1