通过文档学习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点阵,用该属性存储结点信息。