From 0bcfcfd1140623d1f2ef4de2eb12afb5b8bd2566 Mon Sep 17 00:00:00 2001 From: Mohamed haseeb Date: Sat, 23 Oct 2021 17:59:18 +0200 Subject: [PATCH] Speeding up test_voting/test_gridsearch --- sklearn/ensemble/tests/test_voting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/ensemble/tests/test_voting.py b/sklearn/ensemble/tests/test_voting.py index e0b44afd1e004..4bebfaca53709 100644 --- a/sklearn/ensemble/tests/test_voting.py +++ b/sklearn/ensemble/tests/test_voting.py @@ -263,7 +263,7 @@ def test_multilabel(): def test_gridsearch(): """Check GridSearch support.""" clf1 = LogisticRegression(random_state=1) - clf2 = RandomForestClassifier(random_state=1) + clf2 = RandomForestClassifier(random_state=1, n_estimators=3) clf3 = GaussianNB() eclf = VotingClassifier( estimators=[("lr", clf1), ("rf", clf2), ("gnb", clf3)], voting="soft" @@ -275,7 +275,7 @@ def test_gridsearch(): "weights": [[0.5, 0.5, 0.5], [1.0, 0.5, 0.5]], } - grid = GridSearchCV(estimator=eclf, param_grid=params) + grid = GridSearchCV(estimator=eclf, param_grid=params, cv=2) grid.fit(iris.data, iris.target)