win10下pyg之gcn代码成功运行

目录

环境搭建

代码运行

代码准备

解析器选择

数据集准备


环境搭建

本人是基于anaconda创建虚拟环境pyg并安装torch个pyg相关包,安装见

代码运行

本人代码运行使用的是pycharm,安装很简单,可以自行搜索,安装好后进行下面3个步骤

代码准备

直接去pyg官方git下载源代码,然后解压

解析器选择

因为虚拟环境和pycharm已经准备好,因此打开pycharm-文件-open-选择解压后的代码文件见下,选择解析器,具体见本人的另一篇博客解析器选择

打开examples里面很多模型代码,找到下面的gcn.py代码,右击open in terminal,然后就可以像在命令窗口一样执行了,直接输入

python gcn.py

运行下载慢的情况下见

给力的话直接就成功,就不需要下面的数据集准备了

数据集准备

但是事情,并没有上面那么顺利,因为数据集下载及其缓慢,也下载不下来,因此我手动下载了,首先会看到代码自动创建了文件夹(框起来的)

dataset = 'Cora'#目的是zai data下创建Cora目录
'''
osp.join参数分析
osp.realpath(__file__))自动创建路径
.. 是.py文件的上一级,也就是和.py文件的上一级examples同目录
'data' 总文件名
dataset 次文件名(下一级目录)
'''
path = osp.join(osp.dirname(osp.realpath(__file__)), '..', 'data', dataset)自动创建的是data/Cora/Cora/raw,后面的/Cora/raw是Planetoid函数设定的
dataset = Planetoid(path, dataset, transform=T.NormalizeFeatures())
data = dataset[0]

手动下面的8个下载数据集后,放在raw下,再次执行代码就可以正常运行了

运行完后发现代码自动产生了processed文件夹,并生成了一些其他3个文件

数据集具体文件创建是下面gcn代码的第16行确定的,下载代码见17行的函数,该函数定义在pytorch_geometric/torch_geometric/datasets/planetoid.py,也可以在官方查看

正常运行结果见

其他代码运行

同时gcn2_cora.py运行结果如下

gat.py运行结果如下

gnn_explaniner.py运行结果

dataset = 'Cora'
path = osp.join(osp.dirname(osp.realpath(__file__)), '..', 'data', 'Planetoid')#和cora数据集没差,只是名字不一样,把cora下载的raw数据集复制了过来
dataset = Planetoid(path, dataset, transform=T.NormalizeFeatures())
data = dataset[0]

猜你喜欢

转载自blog.csdn.net/lj2048/article/details/111588975