1、简介
数据集下载路径问题
dataset = Planetoid(root='./data/',name='Cora') # ./下载到当前同级下的data文件下(自动创建data目录,具体的Cora下载就是内置函数的事情了)
dataset = TUDataset(root='/data/ENZYMES', name='ENZYMES') # /是直接在F盘下创建data/ENZYMES路径,并使用内置函数处理下载ENZYMES数据
2、代码
'''
PyG手册入门
'''
# 给出了一个具有3个节点和4条边的无权无向图的简单例子,每个节点只包含一个特征
import torch # 基于pytorch的
from torch_geometric.data import Data # 数据处理接口,一般为了初始化节点和邻接矩阵
# 对于一个具有3个节点,4条边的具体简单无权无向图(边的2倍,正反)
'''
需要初始化节点特征x【N,F】和邻接矩阵edge_index【2,E】
N节点数
F节点特征维度数
E边数
注意
X
在这里节点特征x就是节点本身的值,维度就是1了
数据类型:long
edge_index
即定义所有边的源节点和目标节点的张量,并不是索引元组的列表
数据类型:float
'''
#边,边结构信息初始化如下;shape = [2,num_edge]
edge_index = torch.tensor([[0,