torch-geometric

Coordinate end-to-end graph neural network experiments using PyTorch Geometric.

3|Updated Mar 11, 2026
One-click install
npx skills add https://github.com/JunMA98/Computer-science-claude-skills --skill torch-geometric-junma98
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/JunMA98/Computer-science-claude-skills/tree/main/skills/torch-geometric
Command: npx skills add https://github.com/JunMA98/Computer-science-claude-skills --skill torch-geometric-junma98

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

PyTorch Geometric enables researchers to build, train, and evaluate graph neural networks for a variety of CS tasks, but setting up templates, datasets, and benchmarks can be time consuming.

Core Features & Use Cases

  • End-to-end GNN workflows: model templates (GCN, GAT, GraphSAGE), dataset loading, training loops, and evaluation.
  • Task coverage: node classification, graph classification, and link prediction across citation networks, social graphs, and knowledge graphs.
  • Reference material and tooling: comprehensive references for datasets, layers, and transforms, plus scripts to benchmark and visualize graphs.

Quick Start

Train a minimal two-layer GCN on the Cora dataset using PyG to establish a baseline quickly.

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 a graph neural network for node classification using PyTorch Geometric?

To perform node classification with graph neural networks, this Skill provides end-to-end PyTorch Geometric templates including model architectures like GCN, dataset loading scripts, and training loops to quickly establish baselines on citation networks.

Can I run graph classification and link prediction workflows with PyG?

Yes, graph classification and link prediction workflows are supported through coordinated PyTorch Geometric templates. These workflows apply to various domains including social graphs, knowledge graphs, and programming graphs for comprehensive GNN evaluation.

What PyTorch dependencies do I need to set up GNN experiments with PyTorch Geometric?

You need PyTorch, PyTorch Geometric, and NumPy installed to run graph neural network experiments. These libraries provide the foundational tensor computations and graph data structures required for the provided model templates and benchmarking scripts.

How do I benchmark GNN models across different graph datasets?

You can benchmark graph neural network models using the provided scripts and dataset references. The Skill includes benchmarking utilities and visualization tools to evaluate model performance across citation networks and other graph structures.

What's the best way to quickly start training a GCN model on the Cora dataset?

The quickest way to start training a GCN model on the Cora dataset is using the minimal two-layer GCN template provided. This quick start approach allows you to establish a baseline rapidly using PyTorch Geometric.

Does this Skill provide reference material for GNN layers and transforms in PyTorch Geometric?

Yes, comprehensive references for GNN layers, datasets, and transforms are included. These references support the end-to-end graph neural network workflows by detailing the available PyTorch Geometric components for model customization.