使用PyG创建自定义图数据集

在图神经网络中，数据集是训练和评估模型的关键组成部分。PyG（PyTorch Geometric）是一个用于处理图数据的Python库，它提供了创建和处理图数据集的功能。本文将介绍如何使用PyG创建自定义的图数据集，并提供相应的源代码示例。

首先，我们需要安装PyG库。可以使用以下命令通过pip安装最新版本的PyG：

pip install torch-geometric

安装完成后，我们可以开始创建自定义的图数据集。首先，我们需要导入所需的库：

import torch
from torch_geometric.data import Dataset

接下来，我们定义一个继承自torch_geometric.data.Dataset的自定义数据集类。在这个类中，我们需要实现以下方法：

• __len__：返回数据集的大小。
• __getitem__：根据给定的索引返回图数据。

下面是一个示例，展示了如何创建一个简单的自定义图数据集类：