feat: make it harder to rollback from FINAL

AllenNeuralDynamics · Jun 18, 2024 · 8035174 · 8035174
1 parent f03ed18
commit 8035174
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/code/aind_auto_train/curriculums/coupled_baiting_2p2.py b/code/aind_auto_train/curriculums/coupled_baiting_2p2.py
@@ -366,11 +366,11 @@
         TransitionRule(
             decision=Decision.ROLLBACK,
             to_stage=TrainingStage.STAGE_3,
-            condition_description="For recent 2 sessions, mean finished trials < 350 or efficiency < 0.65",
+            condition_description="For recent 5 sessions, mean finished trials < 300 or efficiency < 0.60",
             condition="""lambda metrics:
-                        np.mean(metrics.finished_trials[-2:]) < 350
+                        np.mean(metrics.finished_trials[-5:]) < 300
                         or
-                        np.mean(metrics.foraging_efficiency[-2:]) < 0.65
+                        np.mean(metrics.foraging_efficiency[-5:]) < 0.60
                         """,
         ),
     ]

diff --git a/code/aind_auto_train/curriculums/uncoupled_baiting_2p2.py b/code/aind_auto_train/curriculums/uncoupled_baiting_2p2.py
@@ -352,11 +352,11 @@
         TransitionRule(
             decision=Decision.ROLLBACK,
             to_stage=TrainingStage.STAGE_3,  # Back to C0B0 with auto water
-            condition_description="For recent 2 sessions, mean finished trials < 300 or efficiency < 0.6",
+            condition_description="For recent 5 sessions, mean finished trials < 300 or efficiency < 0.6",
             condition="""lambda metrics:
-                        np.mean(metrics.finished_trials[-2:]) < 300
+                        np.mean(metrics.finished_trials[-5:]) < 300
                         or
-                        np.mean(metrics.foraging_efficiency[-2:]) < 0.6
+                        np.mean(metrics.foraging_efficiency[-5:]) < 0.60
                         """,
         ),
     ]

diff --git a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p2.py b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p2.py
@@ -350,11 +350,11 @@
         TransitionRule(
             decision=Decision.ROLLBACK,
             to_stage=TrainingStage.STAGE_3,  # Back to C0B0 with auto water
-            condition_description="For recent 2 sessions, mean finished trials < 300 or efficiency < 0.6",
+            condition_description="For recent 5 sessions, mean finished trials < 300 or efficiency < 0.6",
             condition="""lambda metrics:
-                        np.mean(metrics.finished_trials[-2:]) < 300
+                        np.mean(metrics.finished_trials[-5:]) < 300
                         or
-                        np.mean(metrics.foraging_efficiency[-2:]) < 0.6
+                        np.mean(metrics.foraging_efficiency[-5:]) < 0.60
                         """,
         ),
     ]

diff --git a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p2rwdDelay159.py b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p2rwdDelay159.py
@@ -414,11 +414,11 @@
         TransitionRule(
             decision=Decision.ROLLBACK,
             to_stage=TrainingStage.STAGE_4,  # Back to C0B0 with auto water
-            condition_description="For recent 2 sessions, mean finished trials < 300 or efficiency < 0.6",
+            condition_description="For recent 5 sessions, mean finished trials < 300 or efficiency < 0.6",
             condition="""lambda metrics:
-                        np.mean(metrics.finished_trials[-2:]) < 300
+                        np.mean(metrics.finished_trials[-5:]) < 300
                         or
-                        np.mean(metrics.foraging_efficiency[-2:]) < 0.6
+                        np.mean(metrics.foraging_efficiency[-5:]) < 0.60
                         """,
         ),
     ]