PyG|实现构建自己的数据集

在torch中构建自己的数据集。

import torch
from torch_geometric.data import Data

x = torch.tensor([[2,1],[5,6],[3,7],[12,0]], dtype=torch.float)
y = torch.tensor([0,1,0,1], dtype=float)

edge_index = torch.tensor([[0,1,2,0,3], # 起始点
                           [1,0,1,3,2]], dtype=torch.long) # 终止点
# 边的顺序打乱都可以


data = Data(x=x, y=y, edge_index=edge_index)

print(data)