visualize_graph
Args: data: PyTorch Geometric Data object title: Plot title node_color_attr: Data attribute to use for node colors (e.g., 'y', 'train_mask') edge_color_attr: Data attribute to use for edge colors show_labels: Whether to show node labels…
- Skills
- Scientific
- 3
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.
- Install the skill (command above).
- In Claude Code, describe what you need - the skill turns on by itself when the context fits.
- No commands to memorize: you talk in plain language and the skill does the rest.
Install (Claude Code)
mkdir -p .claude/skills/visualize-graph && curl -fsSL https://doutorai.pt/data/lib/skills__visualize-graph.md -o .claude/skills/visualize-graph/SKILL.mdPaste this command into the terminal, in your project folder, and the component installs automatically.
Content
#!/usr/bin/env python3
"""
Visualize PyTorch Geometric graph structures using networkx and matplotlib.
This script provides utilities to visualize Data objects, including:
- Graph structure (nodes and edges)
- Node features (as colors)
- Edge attributes (as edge colors/widths)
- Community/cluster assignments
Usage:
python visualize_graph.py --dataset Cora --output graph.png
Or import and use:
from scripts.visualize_graph import visualize_data
visualize_data(data, title="My Graph", show_labels=True)
"""
import argparse
import matplotlib.pyplot as plt
import networkx as nx
import
- 🔒 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.