diff options
Diffstat (limited to 'examples/py-machine-learning/sklearn-random-forest.py')
| -rwxr-xr-x | examples/py-machine-learning/sklearn-random-forest.py | 2 |
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'], |