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

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

作者: 深夏尤飞六出花 | 来源:发表于2022-07-25 21:04 被阅读0次

参考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异构图注意力神经网络算法

      本文链接:https://www.haomeiwen.com/subject/noknirtx.html