Sequence Classification With Attention Visualization And Hidden State Extraction

text-classification model by undefined. 7,25,081 downloads.

Unique: Provides access to intermediate transformer representations (all 12 layer outputs and attention weights) through a unified API, enabling post-hoc interpretability analysis without modifying the model architecture. The SequenceClassifierOutput dataclass exposes these tensors in a structured format compatible with visualization and analysis libraries.

vs others: Enables interpretability analysis without requiring custom model modifications or separate explanation models (e.g., LIME, SHAP), and provides direct access to learned representations compared to black-box APIs.