gnn-advanced-viz - SKILL.md Agent Skill

name: gnn-advanced-viz description: GNN advanced visualization and interactive plots. Use when creating D2 diagrams, dashboards, interactive network visualizations, timeline charts, heatmaps, or data extraction for custom visualizations.

GNN Advanced Visualization (Step 9)

Purpose

Generates advanced visualizations including D2 diagrams, interactive dashboards, network and timeline visualizations, heatmaps, and data extraction for GNN models.

Key Commands

# Run advanced visualization
python src/9_advanced_viz.py --target-dir input/gnn_files --output-dir output --verbose

# As part of pipeline
python src/main.py --only-steps 9 --verbose

API

from advanced_visualization import (
    AdvancedVisualizer,
    create_visualization_from_data, create_network_visualization,
    create_timeline_visualization, create_heatmap_visualization,
    DashboardGenerator, generate_dashboard,
    VisualizationDataExtractor, extract_visualization_data,
    process_advanced_viz,
)

# Create visualizations
viz = AdvancedVisualizer()
create_network_visualization(data, output_path="net.html")
create_heatmap_visualization(data, output_path="heat.html")

# Generate dashboards
gen = DashboardGenerator()
generate_dashboard(model_data, output_dir="output/")

# Extract visualization data
extractor = VisualizationDataExtractor()
viz_data = extract_visualization_data(parsed_model)

# D2 diagram generation (if available)
from advanced_visualization import D2Visualizer, D2_AVAILABLE
if D2_AVAILABLE:
    d2 = D2Visualizer()

Key Exports

AdvancedVisualizer — main advanced visualization class
DashboardGenerator / generate_dashboard — interactive HTML dashboards
VisualizationDataExtractor / extract_visualization_data — data preparation
D2Visualizer / D2DiagramSpec / D2GenerationResult — D2 diagram support
create_network_visualization, create_timeline_visualization, create_heatmap_visualization

Dependencies

# Core interactive visualization
uv sync

# For D2 diagrams (requires d2 CLI)
# Install d2: https://d2lang.com/tour/install

Output

Interactive HTML files in output/9_advanced_viz_output/
D2 diagrams (when D2 is available)
Dashboard pages

MCP Tools

This module registers tools with the GNN MCP server (see mcp.py):

check_visualization_capabilities
get_advanced_visualization_module_info
list_d2_visualization_types
process_advanced_visualization

References

AGENTS.md — Module documentation
README.md — Usage guide
SPEC.md — Module specification

Documentation

README: Module Overview
AGENTS: Agentic Workflows
SPEC: Architectural Specification
SKILL: Capability API