Skip to content

Commit

Permalink
hotfix: decrease foraging eff threshold from 0.75 to 0.70 for graduation
Browse files Browse the repository at this point in the history
To be more conservative...
  • Loading branch information
hanhou committed Apr 15, 2024
1 parent 15bb2eb commit adbfba9
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions code/aind_auto_train/curriculums/coupled_baiting_2p0.py
Original file line number Diff line number Diff line change
Expand Up @@ -351,7 +351,7 @@
decision=Decision.PROGRESS,
to_stage=TrainingStage.GRADUATED,
condition_description=("For recent 5 sessions,"
"mean finished trials >= 500 and mean efficiency >= 0.75 "
"mean finished trials >= 500 and mean efficiency >= 0.70 "
"and total sessions >= 10 and sessions at final >= 5"),
condition="""lambda metrics:
metrics.session_total >= 10
Expand All @@ -360,7 +360,7 @@
and
np.mean(metrics.finished_trials[-5:]) >= 500
and
np.mean(metrics.foraging_efficiency[-5:]) >= 0.75
np.mean(metrics.foraging_efficiency[-5:]) >= 0.70
""",
),
TransitionRule(
Expand Down
4 changes: 2 additions & 2 deletions code/aind_auto_train/curriculums/uncoupled_baiting_2p0.py
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,7 @@
decision=Decision.PROGRESS,
to_stage=TrainingStage.GRADUATED,
condition_description=("For recent 5 sessions,"
"mean finished trials >= 500 and mean efficiency >= 0.75 "
"mean finished trials >= 500 and mean efficiency >= 0.70 "
"and total sessions >= 10 and sessions at final >= 5"),
condition="""lambda metrics:
metrics.session_total >= 10
Expand All @@ -344,7 +344,7 @@
and
np.mean(metrics.finished_trials[-5:]) >= 500
and
np.mean(metrics.foraging_efficiency[-5:]) >= 0.75
np.mean(metrics.foraging_efficiency[-5:]) >= 0.70
""",
),
TransitionRule(
Expand Down
4 changes: 2 additions & 2 deletions code/aind_auto_train/curriculums/uncoupled_no_baiting_2p0.py
Original file line number Diff line number Diff line change
Expand Up @@ -332,7 +332,7 @@
decision=Decision.PROGRESS,
to_stage=TrainingStage.GRADUATED,
condition_description=("For recent 5 sessions,"
"mean finished trials >= 500 and mean efficiency >= 0.75 "
"mean finished trials >= 500 and mean efficiency >= 0.70 "
"and total sessions >= 10 and sessions at final >= 5"),
condition="""lambda metrics:
metrics.session_total >= 10
Expand All @@ -341,7 +341,7 @@
and
np.mean(metrics.finished_trials[-5:]) >= 500
and
np.mean(metrics.foraging_efficiency[-5:]) >= 0.75
np.mean(metrics.foraging_efficiency[-5:]) >= 0.70
""",
),
TransitionRule(
Expand Down

0 comments on commit adbfba9

Please sign in to comment.