如何创建图神经网络数据集？（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)