图神经网络自定义数据

Question

我正在遵循本教程。 https://colab.research.google.com/github/AntonioLonga/PytorchGeometricTutorial/blob/main/Tutorial3/Tutorial3.ipynb 我想在自定义数据集上尝试一下。如何使用我自己的数据集而不是 Cora ？我会喜欢任何想法。

Answer 1

Pytorch Geometric 文档有一个很棒的指南。在那里您可以找到明确的说明，了解您需要覆盖相应父数据集类的哪些方法。这意味着您可能必须编写自己的方法来读取自己的文件格式并将其带入

Data

形状的Pytorch Geometric。

许多用于标准图机器学习任务的常用基准数据集都集成到了 pytorch 几何中，您可以简单地导入它们并使用加载它们（注意，如果您愿意，您甚至可以添加自定义预处理）：

from torch_geometric.datasets import <Some Dataset>

dataset = <Some Dataset>()

# Ready to use the dataset

您可以在https://pytorch-geometric.readthedocs.io/en/2.5.3/modules/datasets.html

找到直接可用的数据集列表