har_trees: Add a dataset definition

emlearn · Jan 2, 2025 · 8bf8ed1 · 8bf8ed1
1 parent ef78bfb
commit 8bf8ed1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/examples/har_trees/har_train.py b/examples/har_trees/har_train.py
@@ -318,6 +318,18 @@ def run_pipeline(run, hyperparameters, dataset,
                 'True', 'False',
             ],
         ),
+        'toothbrush_jonnor': dict(
+            groups=['session'],
+            label_column = 'is_brushing',
+            time_column = 'time',
+            data_columns = ['x', 'y', 'z'],
+            #data_columns = ['gravity_x', 'gravity_y', 'gravity_z'],
+            #data_columns = ['motion_x', 'motion_y', 'motion_z'],
+            classes = [
+                #'mixed',
+                'True', 'False',
+            ],
+        ),
     }
 
     if not dataset in dataset_config.keys():