PredictionsStore

class ethograph.labels.predictions.PredictionsStore(folder)

Bases: object

Read one ethograph.segment.inference prediction folder.

Example

store = PredictionsStore("labels/predictions_mstcn_20260101_000000")
labels_df, _ = store.load_all(dt)
confidence = store.get_confidence(trial=5, dt=dt, individual="A")

Methods

get_confidence(trial, dt[, individual])	Per-frame confidence for one (trial, individual), from the run's own probabilities.
load_all(dt[, individual])	Every trial's predictions, already postprocessed by the run itself.

get_confidence(trial, dt, individual=None)

Per-frame confidence for one (trial, individual), from the run’s own probabilities.

Returns None when the run has no .npz (e.g. a hand-edited folder) or no key matches — an aid to review, never something a caller depends on.

Return type:

np.ndarray | None

load_all(dt, individual=None, **_ignored)

Every trial’s predictions, already postprocessed by the run itself.

Return type:

tuple[DataFrame, dict]