torch-geometric

Develop and train Graph Neural Networks with PyTorch Geometric.

Updated Mar 15, 2026
One-click install
npx skills add https://github.com/sagunkayastha/claude_skills_collection --skill torch-geometric-sagunkayastha
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/sagunkayastha/claude_skills_collection/tree/main/machine-learning-ai/torch-geometric
Command: npx skills add https://github.com/sagunkayastha/claude_skills_collection --skill torch-geometric-sagunkayastha

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

This Skill simplifies the process of building and training Graph Neural Networks (GNNs), enabling you to leverage deep learning on complex graph-structured data.

Core Features & Use Cases

  • GNN Model Development: Build custom GNN layers or use pre-built ones like GCN, GAT, and GraphSAGE.
  • Data Handling: Load and process various graph datasets, including citation networks, molecular graphs, and social networks.
  • Training & Evaluation: Provides templates for node and graph classification tasks.
  • Use Case: Analyze social networks to predict user behavior, predict molecular properties for drug discovery, or classify scientific papers based on citation links.

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 using PyTorch?

You can build and train Graph Neural Networks using PyTorch by utilizing pre-built layers like GCN, GAT, and GraphSAGE, or by constructing custom layers for node and graph classification tasks. It provides data loading, model building, and evaluation pipelines.

What types of graph-structured data can I analyze for deep learning?

You can analyze various graph-structured data, including citation networks, molecular graphs for drug discovery, and social networks to predict user behavior. The framework handles data loading and processing for these complex structures to facilitate deep learning.

How do I implement node and graph classification tasks on heterogeneous graphs?

You can implement node and graph classification tasks on heterogeneous graphs by using the provided training and evaluation templates. The framework supports heterogeneous graph structures alongside advanced features like explainability and pooling.

Can I use custom GNN layers for molecular property prediction?

Yes, you can use custom GNN layers or pre-built ones like GraphSAGE for molecular property prediction. The framework allows you to load molecular graphs and execute training pipelines to predict properties for drug discovery.

Does this framework support graph explainability and pooling?

Yes, the framework supports advanced features including explainability and pooling. These features are integrated alongside custom layer development and heterogeneous graph support to enhance model training and evaluation.

Do I need to install PyTorch separately to use PyTorch Geometric?

Yes, PyTorch is a required dependency. You can install PyTorch Geometric via `uv pip install torch_geometric` to start building models for analyzing complex graph-structured data.