使用PYG创建多图数据集

使用 PyTorch Geometric (PYG) 创建多图数据集需要遵循以下步骤：

1. 导入必要的包

import torch
from torch_geometric.data import Data, Dataset

2. 定义单个图的数据结构

class GraphData(Data):
    def __init__(self, x=None, edge_index=None, y=None):
        super(GraphData, self).__init__()
        self.x = x
        self.edge_index = edge_index
        self.y = y

3. 定义数据集类

class GraphDataset(Dataset):
    def __init__(self, graphs=[]):
        super(GraphDataset, self).__init__()
        self.graphs = graphs

    def len(self):
        return len(self.graphs)

    def get(self, idx):
        return self.graphs[idx]

4. 创建多个图的数据集

graph1 = GraphData(
    x=torch.tensor([[1], [2], [3]], dtype=torch.float),
    edge_index=torch.tensor([[0, 1], [1, 2]], dtype=torch.long).t().contiguous(),
    y=torch.tensor([0], dtype=torch.long)
)
graph2 = GraphData(
    x=torch.tensor([[4], [5], [6]], dtype=torch.float),
    edge_index=torch.tensor([[0, 1], [1, 2]], dtype=torch.long).t().contiguous(),
    y=torch.tensor([1], dtype=torch.long)
)
dataset = GraphDataset(graphs=[graph1, graph2])

这样就可以创建一个包含多个图的数据集。每个图都是一个 `GraphData` 对象，数据集是 `GraphDataset` 对象。在实际使用中，我们可以通过调用 `dataset.get(idx)` 方法来获取数据集中的某个图。