AQIT 0.1.0
Loading...
Searching...
No Matches
aquin.sae_stats_check Namespace Reference

Functions

dict[str, Any] _normalize_sae_stats_result (dict[str, Any] result)
tuple[Path, Path] write_sae_stats_check (dict[str, Any] result, *, str|None tool_name, str|Path cwd)
None _plot_sae_stats_check (dict[str, Any] result, Path png_path)
None _plot_layer_profile (Any ax, list[dict[str, Any]] profile, *, str metric, str title, str color, bool pct=False)
None _plot_heatmap (Any ax, dict[str, Any] heatmap, list[dict[str, Any]] probes)
str _probe_label (str row_id, list[dict[str, Any]] probes)
str _trunc (str text, int limit)

Variables

str _JSON_NAME = "sae-stats-check.json"
str _PNG_NAME = "sae-stats-check.png"

Function Documentation

◆ _normalize_sae_stats_result()

dict[str, Any] _normalize_sae_stats_result ( dict[str, Any] result)
protected

Definition at line 19 of file sae_stats_check.py.

Referenced by write_sae_stats_check().

◆ _plot_heatmap()

None _plot_heatmap ( Any ax,
dict[str, Any] heatmap,
list[dict[str, Any]] probes )
protected

Definition at line 139 of file sae_stats_check.py.

References _probe_label().

Referenced by _plot_sae_stats_check().

◆ _plot_layer_profile()

None _plot_layer_profile ( Any ax,
list[dict[str, Any]] profile,
* ,
str metric,
str title,
str color,
bool pct = False )
protected

Definition at line 111 of file sae_stats_check.py.

Referenced by _plot_sae_stats_check().

◆ _plot_sae_stats_check()

None _plot_sae_stats_check ( dict[str, Any] result,
Path png_path )
protected

Definition at line 69 of file sae_stats_check.py.

References _plot_heatmap(), and _plot_layer_profile().

Referenced by write_sae_stats_check().

◆ _probe_label()

str _probe_label ( str row_id,
list[dict[str, Any]] probes )
protected

Definition at line 173 of file sae_stats_check.py.

References _trunc().

Referenced by _plot_heatmap().

◆ _trunc()

str _trunc ( str text,
int limit )
protected

Definition at line 186 of file sae_stats_check.py.

Referenced by _probe_label().

◆ write_sae_stats_check()

tuple[Path, Path] write_sae_stats_check ( dict[str, Any] result,
* ,
str | None tool_name,
str | Path cwd )

Definition at line 48 of file sae_stats_check.py.

References _normalize_sae_stats_result(), and _plot_sae_stats_check().

Variable Documentation

◆ _JSON_NAME

str aquin.sae_stats_check._JSON_NAME = "sae-stats-check.json"
protected

Definition at line 15 of file sae_stats_check.py.

◆ _PNG_NAME

str aquin.sae_stats_check._PNG_NAME = "sae-stats-check.png"
protected

Definition at line 16 of file sae_stats_check.py.