GCN 学习 Pytorch-Geometric 教程 第二版

PyG教程二

• 数据处理
  • 图卷积网络

数据转换

PyTorch Geometric内置了特定的转换机制，该机制以Data对象为输入，并生成经过处理后的全新Data对象。用户可通过torch_geometric.transforms.Compose实现多个转换操作的串联，并可在数据集保存至存储设备前（pre_transform）或在访问数据集内图结构之前（transform）执行这些转换操作。
以下将通过一个具体案例进行说明：我们针对ShapeNet数据集（该数据集包含17,000个三维形状点云数据，且每个点均带有对应于16个形状类别的标签信息）实施相应的转换处理。

    from torch_geometric.datasets import ShapeNet
    dataset = ShapeNet(root='/tmp/ShapeNet', categories=['Airplane'])
    print(dataset[0])
    >>> Data(pos=[2518, 3], y=[2518])
    
    
    AI写代码python
    
    运行

利用transform技术，将点云数据转换为图结构形式。

    import torch_geometric.tra