LLMVis / utils /head_detection.py

Commit History

  1. Rename L#-H# notation to L#-D# for detector consistency⁠

    cdpearlman +1: Claude Opus 4.6
    deb2071

  1. Attention refactor, better categorization and explanation

    7fa8fb4

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

    b828e58

  1. 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

  1. Feature 5: Implement attention head detection and categorization with visual display

    24b78dc