From 1e2a3c6edbac2b889e3f53d65614440e74a3ebbd Mon Sep 17 00:00:00 2001 From: Christophe Bedetti Date: Wed, 21 Sep 2022 13:28:28 -0400 Subject: [PATCH] Remove sklearn FutureWarning --- polyssifier/poly_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/polyssifier/poly_utils.py b/polyssifier/poly_utils.py index 938ec2e..8a15946 100644 --- a/polyssifier/poly_utils.py +++ b/polyssifier/poly_utils.py @@ -125,14 +125,14 @@ def build_classifiers(exclude, scale, feature_selection, nCols): if 'Decision Tree' not in exclude: classifiers['Decision Tree'] = { 'clf': DecisionTreeClassifier(max_depth=None, - max_features='auto'), + max_features='sqrt'), 'parameters': {}} if 'Random Forest' not in exclude: classifiers['Random Forest'] = { 'clf': RandomForestClassifier(max_depth=None, n_estimators=10, - max_features='auto'), + max_features='sqrt'), 'parameters': {'n_estimators': list(range(5, 20))}} if 'Logistic Regression' not in exclude: