From e19bb7ce5a0c8d1d2a2575d4a3181aec18494509 Mon Sep 17 00:00:00 2001
From: Hanmin Qin <qinhanmin2005@sina.com>
Date: Sun, 21 May 2017 13:35:02 +0800
Subject: [PATCH] Fix n_splits in KFold instantiation in model_selection tests
(#8910)
---
sklearn/model_selection/tests/test_validation.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/sklearn/model_selection/tests/test_validation.py b/sklearn/model_selection/tests/test_validation.py
index 9228837e1b..c05b25ce67 100644
--- a/sklearn/model_selection/tests/test_validation.py
+++ b/sklearn/model_selection/tests/test_validation.py
@@ -921,7 +921,7 @@ def check_cross_val_predict_with_method(est):
X, y = shuffle(X, y, random_state=0)
classes = len(set(y))
- kfold = KFold(len(iris.target))
+ kfold = KFold()
methods = ['decision_function', 'predict_proba', 'predict_log_proba']
for method in methods:
--
GitLab