aquin.feature_logits_check Namespace Reference

Functions
dict[str, Any]	_flatten_feature_logits_result (dict[str, Any] result)
tuple[Path, Path]	write_feature_logits_check (dict[str, Any] result, *, str|None tool_name, str|Path cwd)
None	_plot_feature_logits_check (dict[str, Any] result, Path png_path)
None	_plot_token_panel (Any ax, list[dict[str, Any]] rows, str title, str color, *, bool positive)
str	_short_token (str tok, int limit=16)

Variables
str	_JSON_NAME = "feature-logits-check.json"
str	_PNG_NAME = "feature-logits-check.png"

Function Documentation

_flatten_feature_logits_result()

dict[str, Any] _flatten_feature_logits_result ( dict[str, Any] result )

protected

Definition at line 19 of file feature_logits_check.py.

Referenced by write_feature_logits_check().

_plot_feature_logits_check()

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

protected

Definition at line 59 of file feature_logits_check.py.

References _plot_token_panel().

Referenced by write_feature_logits_check().

_plot_token_panel()

None _plot_token_panel ( Any ax, list[dict[str, Any]] rows, str title, str color, * , bool positive )

protected

Definition at line 94 of file feature_logits_check.py.

References _short_token().

Referenced by _plot_feature_logits_check().

_short_token()

str _short_token ( str tok, int limit = 16 )

protected

Definition at line 131 of file feature_logits_check.py.

Referenced by _plot_token_panel().

write_feature_logits_check()

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

Definition at line 38 of file feature_logits_check.py.

References _flatten_feature_logits_result(), and _plot_feature_logits_check().

Variable Documentation

_JSON_NAME

str aquin.feature_logits_check._JSON_NAME = "feature-logits-check.json"

protected

Definition at line 15 of file feature_logits_check.py.

_PNG_NAME

str aquin.feature_logits_check._PNG_NAME = "feature-logits-check.png"

protected

Definition at line 16 of file feature_logits_check.py.