torch-geometric

SKILL.md

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.

Réseaux de neurones graphiques (PyG). Classification de nœuds/graphes, prédiction de liens, GCN, GAT, GraphSAGE, graphiques hétérogènes, prédiction de propriétés moléculaires, pour l'apprentissage profond géométrique. Source : jackspace/claudeskillz.

Voir l'original