From a4eb68cb8b32fa101a390102f45b868ed2cd37d1 Mon Sep 17 00:00:00 2001 From: Guillaume Lemaitre Date: Tue, 25 Apr 2017 10:47:43 +0200 Subject: [PATCH] FIX set min_impurity_split to None in gradient boosting estimator --- sklearn/ensemble/gradient_boosting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/ensemble/gradient_boosting.py b/sklearn/ensemble/gradient_boosting.py index 2c18d338dc715..4db93f0a51711 100644 --- a/sklearn/ensemble/gradient_boosting.py +++ b/sklearn/ensemble/gradient_boosting.py @@ -1433,7 +1433,7 @@ def __init__(self, loss='deviance', learning_rate=0.1, n_estimators=100, subsample=1.0, criterion='friedman_mse', min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0., max_depth=3, min_impurity_decrease=0., - min_impurity_split=1e-7, init=None, + min_impurity_split=None, init=None, random_state=None, max_features=None, verbose=0, max_leaf_nodes=None, warm_start=False, presort='auto'): @@ -1840,7 +1840,7 @@ def __init__(self, loss='ls', learning_rate=0.1, n_estimators=100, subsample=1.0, criterion='friedman_mse', min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0., max_depth=3, min_impurity_decrease=0., - min_impurity_split=1e-7, init=None, random_state=None, + min_impurity_split=None, init=None, random_state=None, max_features=None, alpha=0.9, verbose=0, max_leaf_nodes=None, warm_start=False, presort='auto'):