From 0ccc5be66d810563018b1e5a07fbf46720f5e2f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Thu, 25 May 2023 23:19:57 +0200
Subject: [PATCH] special labels
---
src/behavior_model/data/larva_dataset.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/behavior_model/data/larva_dataset.py b/src/behavior_model/data/larva_dataset.py
index ae35bd3..2a157aa 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
--
GitLab