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.

그래프 신경망(PyG). 기하학적 딥러닝을 위한 노드/그래프 분류, 링크 예측, GCN, GAT, GraphSAGE, 이종 그래프, 분자 특성 예측. 출처: jackspace/claudeskillz.

원본 보기