Module visualization

Attention and feature visualization for neural network interpretation

This module provides visualization capabilities for understanding neural network behavior including attention visualization, feature visualization, and network dissection.

Structs

AttentionVisualizer

Attention visualizer for transformer models

NetworkDissectionResult

Network dissection result

VisualizationResult

Visualization result containing processed data

Enums

AttentionAggregation

Attention aggregation strategy

VisualizationMethod

Feature visualization method

Functions

create_attention_heatmap

Create attention heatmap for visualization

generate_feature_visualization

Generate feature visualization using specified method

perform_network_dissection

Perform network dissection analysis