"""
Configuration schema for the Analysis mode.
"""
from dataclasses import dataclass
from omegaconf import MISSING
[docs]
@dataclass
class AnalysisConfig:
"""
Configuration for Deep Learning Analysis.
Parameters
----------
enable_internals : bool
Whether to analyze model internals (params, stats).
enable_dynamics : bool
Whether to analyze training dynamics (gradients).
enable_errors : bool
Whether to analyze prediction errors (confusion matrix).
batch_size : int
Batch size for dynamic analysis.
output_dir : str
Directory to save analysis results.
save_plots : bool
Whether to save visualization plots.
save_json : bool
Whether to save raw results as JSON.
ckpt_path : str
Path to the model checkpoint to load.
"""
enable_internals: bool = True
enable_dynamics: bool = False
enable_errors: bool = True
batch_size: int = 4
output_dir: str = "analysis_outputs"
save_plots: bool = True
save_json: bool = True
ckpt_path: str = MISSING