aquin.sdk.data Namespace Reference

Functions
list[dict[str, Any]]	_load_probes (Any probes)
dict[str, Any]	capture_activations (*, str|Path output_dir, Any probes=None, str|None model_id=None, **Any kwargs)
dict[str, Any]	replay_activations (str|Path path, *, str|None filter_expr=None, str|None group_by=None, int|None limit=None)
dict[str, Any]	features_compare (**Any kwargs)

Function Documentation

_load_probes()

list[dict[str, Any]] _load_probes ( Any probes )

protected

Definition at line 21 of file data.py.

Referenced by capture_activations().

capture_activations()

dict[str, Any] capture_activations	(	*	,
		str | Path	output_dir,
		Any	probes = None,
		str | None	model_id = None,
		**Any	kwargs )

Record activations (+ metadata) into ``output_dir``.

Definition at line 45 of file data.py.

References _load_probes().

features_compare()

dict[str, Any] features_compare

(

**Any

kwargs

)

Definition at line 95 of file data.py.

replay_activations()

dict[str, Any] replay_activations	(	str | Path	path,
		*	,
		str | None	filter_expr = None,
		str | None	group_by = None,
		int | None	limit = None )

Slice a saved capture by metadata without re-running the model.

Definition at line 74 of file data.py.