torch-geometric

PyTorch Geometric is a library built on PyTorch for developing and training Graph Neural Networks (GNNs). Apply this skill for deep learning on graphs and irregular structures, including mini-batch processing, multi-GPU training, and geometric deep learning applications.

PyG represents graphs using the torchgeometric.data.Data class with these key attributes:

Important: These attributes are not mandatory—extend Data objects with custom attributes as needed.

图神经网络（PyG）。节点/图分类、链接预测、GCN、GAT、GraphSAGE、异构图、分子属性预测、用于几何深度学习。 来源：ovachiever/droid-tings。