ibl_bwb_eval.scoring.ts2#

TS2 scoring utility.

Loads prediction .safetensors files, aligns them to pre-generated ground truth files by window_timestamps, and computes Poisson D² and bits-per-spike.

Usage:

python -m ibl_bwb_eval.scoring.ts2 –pred_dir predictions/<label> –gt_dir ground_truth/ts2 python -m ibl_bwb_eval.scoring.ts2 –pred_dir predictions/<label>/my-model/ts2-co_smoothing –gt_dir ground_truth/ts2

score_file

Score one prediction file against its ground truth file.

score_dir

Score all TS2 prediction files found under pred_dir.

summarize

Aggregate per-seed scores into mean ± SEM per (label, task, recording_id).