diff --git a/sklearn/model_selection/_split.py b/sklearn/model_selection/_split.py index de889fab0bda5cfffb645145edfefc9d622f7ddc..369125f339559a67d82488bbeda141fa70a7ad8c 100644 --- a/sklearn/model_selection/_split.py +++ b/sklearn/model_selection/_split.py @@ -958,8 +958,8 @@ class _RepeatedSplits(with_metaclass(ABCMeta)): if not isinstance(n_repeats, (np.integer, numbers.Integral)): raise ValueError("Number of repetitions must be of Integral type.") - if n_repeats <= 1: - raise ValueError("Number of repetitions must be greater than 1.") + if n_repeats <= 0: + raise ValueError("Number of repetitions must be greater than 0.") if any(key in cvargs for key in ('random_state', 'shuffle')): raise ValueError( diff --git a/sklearn/model_selection/tests/test_split.py b/sklearn/model_selection/tests/test_split.py index e97fdce5e1e5aaec7d72540998a819c3fd3f11c8..546609f413c1549939c02583cf3e3ad01c17097f 100644 --- a/sklearn/model_selection/tests/test_split.py +++ b/sklearn/model_selection/tests/test_split.py @@ -832,9 +832,9 @@ def test_leave_one_p_group_out_error_on_fewer_number_of_groups(): def test_repeated_cv_value_errors(): - # n_repeats is not integer or <= 1 + # n_repeats is not integer or <= 0 for cv in (RepeatedKFold, RepeatedStratifiedKFold): - assert_raises(ValueError, cv, n_repeats=1) + assert_raises(ValueError, cv, n_repeats=0) assert_raises(ValueError, cv, n_repeats=1.5)