Does this skill support heterogeneous graphs?

Yes, it provides guidance on using HeteroData objects to represent graphs with different types of nodes and edges, often found in knowledge graphs and recommender systems.

Can this skill help with large-scale graph datasets?

Yes, it includes patterns for neighbor sampling, distributed training, and managing datasets that exceed available memory by using the Dataset class instead of InMemoryDataset.

What is the primary advantage of using PyTorch Geometric?

PyG provides a highly efficient framework for deep learning on graphs, supporting sparse operations, mini-batching through block-diagonal matrices, and a wide array of pre-implemented research layers.

How does PyG handle batching differently than standard PyTorch?

Unlike standard tensors that require padding, PyG stacks adjacency matrices diagonally and concatenates node features, creating one large graph where each node is mapped back to its original graph via a batch vector.

PyTorch Geometric (PyG)

Name: PyTorch Geometric (PyG)
Author: Weiwei-Mao

byWeiwei-Mao

0•

数据科学与机器学习

Develops and trains Graph Neural Networks (GNNs) for node classification, link prediction, and geometric deep learning tasks.

The PyTorch Geometric (PyG) skill equips Claude with the specialized knowledge required to build, train, and optimize Graph Neural Networks on irregular data structures. It provides comprehensive patterns for implementing state-of-the-art architectures like Graph Convolutional Networks (GCN), Graph Attention Networks (GAT), and GraphSAGE. This skill is particularly valuable for developers working on molecular property prediction, social network analysis, and recommendation systems, as it covers everything from efficient mini-batch processing and message-passing paradigms to handling large-scale heterogeneous graphs and custom data loading.

主要功能

01Seamless integration with benchmark datasets like Cora, QM9, and TUDataset

02Support for heterogeneous graphs with multi-type nodes and edges

030 GitHub stars

04Implementation patterns for 40+ GNN layers including GCN, GAT, and GraphSAGE

05Custom message-passing layer construction using the MessagePassing base class

06Efficient mini-batch processing via block-diagonal adjacency matrix concatenation

使用场景

01Predicting molecular properties and drug-target interactions in bioinformatics

02Building community detection and influence prediction models for social networks

03Analyzing 3D geometric data such as point clouds and molecular structures

What are Skills?·How to Install

Install with 🐟 Skill.Fish

npx skillfish add weiwei-mao/hydrology-skills torch_geometric

For use in Claude.ai and ChatGPT

Download Skill