LLMVis / utils /head_detection.py

Commit History

Attention refactor, better categorization and explanation
7fa8fb4

cdpearlman commited on

Improve explanations: optimize head categorization and comparison thresholds, enhance ablation explanation, rewrite all tooltips for high school level
b828e58

cdpearlman commited on

Refactor: Move attention head categorization to on-demand per-layer display - Removed automatic Attention Head Categorization section from main panel - Created categorize_single_layer_heads() for per-layer categorization - Updated run_analysis to skip automatic head categorization - Enhanced node click callback to show layer-specific categorization - Analysis Results section now displays both BertViz and categorized heads on node click
17ec583

cdpearlman commited on

Feature 5: Implement attention head detection and categorization with visual display
24b78dc

cdpearlman commited on