Pytorch Geometric(PyG)入门

PyG (PyTorch Geometric) 是建立在 PyTorch 基础上的一个库，用于轻松编写和训练图形神经网络 (GNN)，适用于与结构化数据相关的各种应用。官方文档

Install PyG

PyG适用于python3.8-3.12
一般使用场景：pip install torch_geometric 或conda install pyg -c pyg

Get Started

PyG 具有以下主要功能：

• Data Handling of Graphs
• Common Benchmark Datasets
• Mini-batches
• Data Transforms
• Learning Methods on Graphs
• Exercises

Data Handling of Graphs

PyG 中的单个图由 torch_geometric.data.Data 的一个实例描述，默认情况下该实例拥有以下属性：

• data.x: Node feature matrix with shape [num_nodes, num_node_features]
• data.edge_index: Graph connectivity in COO format with shape [2, num_edges] and type torch.long
• data.edge_attr: Edge feature matrix with shape [num_edges, num_edge_features]
• data.y: Target to train against (may have arbitrary shape), e.g., node-level targets of shape [num_nodes, *] or graph-level targets of shape [1, *]
• data.pos: Node position matrix with shape [num_nodes, num_dimensions]

Colab Notebooks and Video Tutorials

官方文档
Pytroch Geometric Tutorials

Tutorials 1

理解一个节点出发的计算图，理解多次计算图后可能节点信息就包含整个图数据信息了，反而没有用。
对应whl地址

安装torch版本对应的pyg，如下所示：

import os
import torch
os.environ['TORCH']