图神经网络实践00--参考PYG文档

通过文档学习PYG

GET STARTED

通过例子介绍模块

图数据处理

模块torch_geometric.data.Data包含了最底层的图数据表示类。用Data类就可以存组织储所有和图相关的信息。

import torch
import torch_geometric.data import data

edge_index = torch.tensor([0, 1, 1, 2],
					      [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = Data(x=x, edge_index=edge_index)
>>>Data(edge_index=[2, 4], x=[3, 1])

注意边的特殊组织形式。下面介绍data的一些属性

data.x
#node feature matrix with shape[num_nodes, num_node_features]
data.edge_index
#Graph connectivity in COO format with shape[2, num_edges]
#note: 注意边连通信息的表达形式较为特殊
data.edge_attr
#Edge feature matrix with shape[num_edges, num_edge_features]
data.y
#Target to train against
#shape--node task:[num_nodes, *] \ graph task:[1, *]
data.pos
#Node position matrix with shape[num_nudes, num_dimensions]
#Note: 一些数据集是3D点阵，用该属性存储结点信息。

上次编辑：2024.3.20