torch_geometric

Develop and train Graph Neural Networks using the PyTorch Geometric library.

8|Updated Jan 13, 2026
One-click install
npx skills add https://github.com/hxk622/TokenDance --skill torch-geometric-hxk622
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch_geometric
Source: https://github.com/hxk622/TokenDance/tree/main/backend/app/skills/builtin/scientific/data-science/torch_geometric
Command: npx skills add https://github.com/hxk622/TokenDance --skill torch-geometric-hxk622

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) and scripts (resource) components.

What problem does it solve?

This Skill simplifies the development and training of Graph Neural Networks (GNNs) for deep learning on graph-structured data, enabling complex analysis of interconnected information.

Core Features & Use Cases

  • GNN Model Building: Implement and train various GNN architectures like GCN, GAT, and GraphSAGE.
  • Data Handling: Load and process diverse graph datasets, including citation networks, molecular graphs, and social networks.
  • Advanced Features: Supports heterogeneous graphs, message passing, pooling, and explainability.
  • Use Case: Analyze a social network to predict user behavior or predict molecular properties for drug discovery.

Quick Start

Install PyTorch Geometric by running uv pip install torch_geometric.

Frequently Asked Questions about torch_geometric

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I build and train Graph Neural Networks for node classification using PyTorch?

Build and train Graph Neural Networks for node classification by implementing architectures like GCN and GraphSAGE using the PyTorch Geometric library. This Skill facilitates loading citation networks and processing graph-structured data to classify nodes based on their graph neighborhood.

What is the best way to predict molecular properties for drug discovery using graph data?

Predict molecular properties for drug discovery by loading molecular graphs and training Graph Neural Networks. The Skill enables graph classification tasks on diverse graph datasets, mapping molecular structures to property predictions using deep learning.

Can I use PyTorch Geometric to handle heterogeneous graphs and message passing?

PyTorch Geometric supports advanced Graph Neural Network concepts including heterogeneous graphs and message passing. You can process interconnected information across diverse node and edge types to analyze complex graph-structured data relationships.

Does PyTorch Geometric support link prediction and explainability tools for graph analysis?

PyTorch Geometric supports link prediction and provides explainability tools for graph analysis. You can train GNN models to predict connections between nodes and utilize built-in tools to interpret the model's decision-making process.

How do I install PyTorch Geometric to start developing GNN models?

Install PyTorch Geometric by running `uv pip install torch_geometric` to set up the environment. This enables the development and training of various GNN architectures for deep learning on graph-structured data.