diff --git a/src/behavior_model/data/larva_dataset.py b/src/behavior_model/data/larva_dataset.py index ae35bd30522e9bdbcea705c64f3f0da860837ac2..2a157aa612dc6708cd1b1405b4d5b4e30226e2f9 100644 --- a/src/behavior_model/data/larva_dataset.py +++ b/src/behavior_model/data/larva_dataset.py @@ -96,6 +96,10 @@ def import_label(s): return 'ROLL' elif s == 'RUN_LARGE': return 'RUN' + elif s == '¬HUNCH': + return 'RUN' # whatever action other than hunch + elif s == '¬ROLL': + return 'RUN' # idem else: return s