From 8035174cfab69f928be57844d7cb00b8c34b5dd3 Mon Sep 17 00:00:00 2001 From: han-lab Date: Tue, 18 Jun 2024 16:59:26 -0700 Subject: [PATCH] feat: make it harder to rollback from FINAL --- code/aind_auto_train/curriculums/coupled_baiting_2p2.py | 6 +++--- code/aind_auto_train/curriculums/uncoupled_baiting_2p2.py | 6 +++--- .../aind_auto_train/curriculums/uncoupled_no_baiting_2p2.py | 6 +++--- .../curriculums/uncoupled_no_baiting_2p2rwdDelay159.py | 6 +++--- 4 files changed, 12 insertions(+), 12 deletions(-) diff --git a/code/aind_auto_train/curriculums/coupled_baiting_2p2.py b/code/aind_auto_train/curriculums/coupled_baiting_2p2.py index b797d1e..00f35a8 100644 --- 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 index 396a9d1..bc548df 100644 --- 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 index ea1004a..cfa87a4 100644 --- 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 index d0bf6fe..64cdf73 100644 --- 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 """, ), ]