Skip to content

Commit 8bf8ed1

Browse files
committed
har_trees: Add a dataset definition
1 parent ef78bfb commit 8bf8ed1

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

examples/har_trees/har_train.py

+12
Original file line numberDiff line numberDiff line change
@@ -318,6 +318,18 @@ def run_pipeline(run, hyperparameters, dataset,
318318
'True', 'False',
319319
],
320320
),
321+
'toothbrush_jonnor': dict(
322+
groups=['session'],
323+
label_column = 'is_brushing',
324+
time_column = 'time',
325+
data_columns = ['x', 'y', 'z'],
326+
#data_columns = ['gravity_x', 'gravity_y', 'gravity_z'],
327+
#data_columns = ['motion_x', 'motion_y', 'motion_z'],
328+
classes = [
329+
#'mixed',
330+
'True', 'False',
331+
],
332+
),
321333
}
322334

323335
if not dataset in dataset_config.keys():

0 commit comments

Comments
 (0)