AQIT 0.1.0
Loading...
Searching...
No Matches
attention_check.py File Reference

Go to the source code of this file.

Namespaces

namespace  aquin
namespace  aquin.attention_check

Functions

tuple[Path, Path] aquin.attention_check.write_attention_check (dict[str, Any] result, *, str|None tool_name, str|Path cwd)
None aquin.attention_check._plot_attention_check (dict[str, Any] result, str|None tool_name, Path png_path)
None aquin.attention_check._plot_llm_attention (dict[str, Any] result, Path png_path, Any plt, Any np)
str aquin.attention_check._short_token (str tok, int limit=10)

Variables

str aquin.attention_check._JSON_NAME = "attention-check.json"
str aquin.attention_check._PNG_NAME = "attention-check.png"