Add more LGBM models to default config

joclement · Mar 5, 2021 · 40a6022 · 40a6022
1 parent e801b83
commit 40a6022
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/config/default.yml b/config/default.yml
@@ -31,6 +31,8 @@ models-to-run:
   - lgbm-finger_ott
   - lgbm-finger_own
   - lgbm-finger_tugraz
+  - lgbm-finger_all
+  - lgbm-tsfresh
   - knn-finger_ott
   - knn-finger_own
   - knn-finger_tugraz
@@ -382,6 +384,16 @@ models:
     classifier:
       n_jobs: *n_jobs
 
+  lgbm-finger_all:
+    normalize: "none"
+    classifier:
+      min_split_gain: 0.1
+      feature_fraction: 1.0
+      min_child_samples: 30
+      max_depth: 50
+      num_leaves: 10
+      n_jobs: *n_jobs
+
   svm-finger_ott:
     normalize: "zscore"
     classifier: