aboutsummaryrefslogtreecommitdiff
path: root/examples/py-machine-learning/sklearn-random-forest.py
diff options
context:
space:
mode:
authorToni Uhlig <matzeton@googlemail.com>2023-06-01 18:55:43 +0200
committerToni Uhlig <matzeton@googlemail.com>2023-06-12 19:05:33 +0200
commitc8ec505b9cebc444bea414ec27c34510ef167bad (patch)
treea38f9ba4edb56021d3d366e801b36dc5a4120151 /examples/py-machine-learning/sklearn-random-forest.py
parent2b1db0a556c31f15e2fcff0c127781fb91c0b402 (diff)
bump libnDPI to 8ea0eaa0d0c4a3be05f67ef7fa1d22c2579cf7d1
* added build fix for Gitlab CI * added friendly C11 check * set required libnDPI versionto 4.7 (ArchLinux ndpi-git sets version to 4.7, which is not released yet) * reduced sklearn-random-forest memory consumption by adjusting min. sample leaf Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
Diffstat (limited to 'examples/py-machine-learning/sklearn-random-forest.py')
-rwxr-xr-xexamples/py-machine-learning/sklearn-random-forest.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/py-machine-learning/sklearn-random-forest.py b/examples/py-machine-learning/sklearn-random-forest.py
index 275376ba2..9716f2b83 100755
--- a/examples/py-machine-learning/sklearn-random-forest.py
+++ b/examples/py-machine-learning/sklearn-random-forest.py
@@ -206,7 +206,7 @@ if __name__ == '__main__':
help='Number of sklearn processes during training.')
argparser.add_argument('--sklearn-estimators', action='store', type=int, default=1000,
help='Number of trees in the forest.')
- argparser.add_argument('--sklearn-min-samples-leaf', action='store', type=int, default=5,
+ argparser.add_argument('--sklearn-min-samples-leaf', action='store', type=int, default=0.0001,
help='The minimum number of samples required to be at a leaf node.')
argparser.add_argument('--sklearn-class-weight', default='balanced', const='balanced', nargs='?',
choices=['balanced', 'balanced_subsample'],