Cookies

With your permission, we use Google Analytics cookies to measure visits and improve the site. Necessary cookies are always on. You can change your choice or withdraw consent anytime in the footer, under Cookie settings. Cookie Policy

Skip to content

create_gnn_template

This script creates ready-to-use GNN model templates with training loops, evaluation metrics, and proper data handling.

  • Skills
  • Scientific
  • 1

What it does and how to use it

A skill that teaches Claude Code to do scientific tasks well. It is packaged knowledge the model uses whenever it is relevant.

  1. Install the skill (command above).
  2. In Claude Code, describe what you need - the skill turns on by itself when the context fits.
  3. No commands to memorize: you talk in plain language and the skill does the rest.

Install (Claude Code)

mkdir -p .claude/skills/create-gnn-template && curl -fsSL https://doutorai.pt/data/lib/skills__create-gnn-template.md -o .claude/skills/create-gnn-template/SKILL.md

Paste this command into the terminal, in your project folder, and the component installs automatically.

Content

#!/usr/bin/env python3
"""
Generate boilerplate code for common GNN architectures in PyTorch Geometric.

This script creates ready-to-use GNN model templates with training loops,
evaluation metrics, and proper data handling.

Usage:
    python create_gnn_template.py --model gcn --task node_classification --output my_model.py
    python create_gnn_template.py --model gat --task graph_classification --output graph_classifier.py
"""

import argparse
from pathlib import Path


TEMPLATES = {
    'node_classification': {
        'gcn': '''import torch
import torch.nn.functional as F
from torch_geome

- 🔒 Full content on the Pro plan (€7.99/month) -

Need help with Claude Code?

Ask Dr AI - it explains things simply.

Need a tailor-made solution?

Not sure where to start with AI in your business? Get a free assessment and a tailored plan - no commitment.

Save hours every week with AI, no hassle

Doutor AI gives you 1,859 ready-to-copy AI prompts, by industry and in plain English - plus a complete technical library for those who want to go further.

See plans