GoogleColab实现HAN异构图注意力神经网络算法

参考dgl代码:

https://github.com/dmlc/dgl/tree/master/examples/pytorch/han

代码说明有数据集和使用,下面使用GoogleColab来实现。

1.整理dgl代码,上传到google drive。

2.打开一个GoogleColab笔记,挂载dgl代码。

from google.colab import drive

drive.mount("/content/drive")

import os

path="/content/drive/My Drive/HAN_test"

os.chdir(path)

3.在笔记里正常输入命令,运行dgl代码文件。

pip install dgl-cu110

你可能感兴趣的:(GoogleColab实现HAN异构图注意力神经网络算法)