Expand description
§tsai_explain
Explainability tools for tsai-rs: activation/gradient capture, attribution maps.
This crate provides:
- Activation capture hooks
- Gradient capture hooks
- Attribution maps (CAM-like for CNNs, attention for transformers)
Structs§
- Activation
Capture - Captured activations from model layers.
- Attribution
Map - Attribution map over time series.
- Gradient
Capture - Captured gradients from model layers.
- Integrated
Gradients Config - Configuration for Integrated Gradients.
Enums§
- Attention
Aggregation - How to aggregate attention weights across layers.
- Attribution
Method - Method for computing attribution maps.
- Baseline
Type - Type of baseline to use for Integrated Gradients.
Functions§
- attention_
attribution - Compute Attention-based attribution for transformer models.
- grad_
cam - Compute GradCAM attribution for CNN models.
- input_
gradient - Compute Input × Gradient attribution.
- integrated_
gradients - Compute Integrated Gradients attribution.
- random_
baseline - Create a random baseline for Integrated Gradients.
- zero_
baseline - Create a zero baseline for Integrated Gradients.