sae_catalog_metrics.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.sae_catalog_metrics

Functions
list[dict[str, Any]]	aquin.compute.sae_catalog_metrics.compute_umap_points_from_sae (str|Path sae_path)
list[tuple[int, float, str]]	aquin.compute.sae_catalog_metrics._collect_top_features (*, str model_id, int layer, list[str] prompts, int n_per_prompt, int max_features)
dict[str, Any]	aquin.compute.sae_catalog_metrics.compute_layer_interp_score (*, str model_id, int layer, str|Path sae_path, list[str]|None prompts=None, int n_features=8, int n_per_prompt=5, int n_samples=8, Any|None client=None)
tuple[Path, Path]	aquin.compute.sae_catalog_metrics.write_catalog_sidecars (*, str|Path sae_path, list[dict[str, Any]] umap_points, dict[str, Any] metrics)
dict[str, Any]	aquin.compute.sae_catalog_metrics.compute_and_write_catalog_sidecars (*, str model_id, int layer, str|Path sae_path, bool skip_interp=False, **Any interp_kwargs)

Variables
list	aquin.compute.sae_catalog_metrics.CATALOG_PROBE_PROMPTS