protocol: freeze oral-B-v2 cold-start recovery

1 files changed, 44 insertions, 0 deletions

diff --git a/sdil/bci_v2_recovery_metrics.py b/sdil/bci_v2_recovery_metrics.py
new file mode 100644
index 0000000..78d3449
--- /dev/null
+++ b/sdil/bci_v2_recovery_metrics.py
@@ -0,0 +1,44 @@
+"""Target-ladder wrapper for the oral-B-v2 cold-start recovery."""
+
+from sdil.bci_v2_metrics import (
+    annotate_events,
+    signature_metrics_v2,
+)
+
+
+def annotate_target_events(
+    events,
+    day,
+    final_success,
+    episode_offset,
+    target_index,
+):
+    """Reuse paired grouping while encoding the fixed target-ladder index."""
+    return annotate_events(
+        events,
+        day,
+        final_success,
+        episode_offset,
+        target_index,
+    )
+
+
+def recovery_signature_metrics(
+    train_events,
+    challenge_mode_events,
+    cfg,
+    role,
+    targets,
+):
+    values = signature_metrics_v2(
+        train_events,
+        challenge_mode_events,
+        cfg,
+        role,
+    )
+    by_index = values.pop("horizon_success_fraction")
+    values["target_success_fraction"] = {
+        str(target): by_index[str(index)]
+        for index, target in enumerate(targets, start=1)
+    }
+    return values