torch-geometric

Build and train Graph Neural Networks for node classification, link prediction, and graph classification using PyTorch Geometric.

Updated Jun 30, 2026
One-click install
npx skills add https://github.com/SunS1eep1ng/codex-dotfiles --skill torch-geometric-suns1eep1ng
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/SunS1eep1ng/codex-dotfiles/tree/main/skills/torch-geometric
Command: npx skills add https://github.com/SunS1eep1ng/codex-dotfiles --skill torch-geometric-suns1eep1ng

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires torch, torch_geometric, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill addresses the challenges of building and training complex Graph Neural Networks (GNNs) for tasks such as node classification, link prediction, and graph classification using PyTorch Geometric, simplifying the implementation of GNNs.

Core Features & Use Cases

  • Graph Data Structures: Efficiently represents graphs with various types and attributes.
  • Diverse GNN Layers: Provides a variety of GNN layers such as GCN, GAT, SAGE, GIN, and more, for different types of GNNs.
  • Dataset Integration: Supports integration with various datasets like Planetoid, TUDataset, and more, simplifying data preparation.
  • Use Case: A researcher can use this Skill to quickly implement a GNN model for node classification on a dataset like Cora.

Quick Start

Load the PyG library with 'uv add torch_geometric' and use it to create a graph with 'Data(x, edge_index, ...)' and then define and train a GNN model using 'torch_geometric.nn...' commands.

Frequently Asked Questions about torch-geometric

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

FAQPage Schema
How do I build a Graph Neural Network for node classification using PyTorch?

To build a Graph Neural Network for node classification, use PyTorch Geometric to create graph data structures and define models using diverse GNN layers like GCN or GAT. This Skill simplifies implementing GNNs for tasks on datasets like Cora.

What is the best way to handle large-scale graph data structures for graph learning?

Handling large-scale graph data structures for graph learning is optimized by using PyTorch Geometric. It efficiently represents graphs with various types and attributes, supporting both homogeneous and heterogeneous graphs for complex graph data processing.

Can I use PyTorch Geometric for link prediction and graph classification tasks?

Yes, you can use PyTorch Geometric for link prediction and graph classification tasks. This Skill builds and trains Graph Neural Networks across these specific tasks, providing diverse layers like SAGE and GIN for different graph learning requirements.

Does this Skill support heterogeneous graphs and standard datasets like TUDataset?

This Skill supports both homogeneous and heterogeneous graphs, and integrates with standard datasets like TUDataset and Planetoid. This simplifies data preparation when building and training Graph Neural Networks using PyTorch Geometric.

Do I need to install PyTorch before using this Skill for graph learning?

Yes, you need to install PyTorch and the PyG library before using this Skill for graph learning. You can load the PyG library with 'uv add torch_geometric' to start defining and training Graph Neural Network models.

What GNN layers are available for implementing graph learning models?

Available GNN layers for implementing graph learning models include GCN, GAT, SAGE, and GIN. These diverse layers allow you to build different types of Graph Neural Networks tailored for node classification, link prediction, and graph classification.