python - 如何创建图神经网络数据集？ (pytorch 几何)

Question

如何将我自己的数据集转换为可供 pytorch 几何图形神经网络使用的数据集？
所有教程都使用已转换为 pytorch 可用的现有数据集。例如，如果我有自己的点云数据集，我如何使用它来训练图神经网络的分类？我自己的用于分类的图像数据集呢？

Answer 1

您需要如何转换数据取决于您的模型期望的格式。
图神经网络通常期望(的子集):

节点特性

边缘

边缘属性

节点目标

取决于问题。您可以在 PyTorch Geometric 中使用这些值的张量创建对象(并根据需要扩展属性)。带 Data像这样的对象:

data = Data(x=x, edge_index=edge_index, y=y)
data.train_idx = torch.tensor([...], dtype=torch.long)
data.test_mask = torch.tensor([...], dtype=torch.bool)