modify kbins test using kmeans due to unstable local minimum (#12450)

scikit-learn · Nov 14, 2018 · fa0e3f2 · fa0e3f2
1 parent a57ef83
commit fa0e3f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sklearn/preprocessing/tests/test_discretization.py b/sklearn/preprocessing/tests/test_discretization.py
@@ -189,10 +189,10 @@ def test_invalid_strategy_option():
 @pytest.mark.parametrize(
     'strategy, expected_2bins, expected_3bins',
     [('uniform', [0, 0, 0, 0, 1, 1], [0, 0, 0, 0, 2, 2]),
-     ('kmeans', [0, 0, 0, 0, 1, 1], [0, 1, 1, 1, 2, 2]),
+     ('kmeans', [0, 0, 0, 0, 1, 1], [0, 0, 1, 1, 2, 2]),
      ('quantile', [0, 0, 0, 1, 1, 1], [0, 0, 1, 1, 2, 2])])
 def test_nonuniform_strategies(strategy, expected_2bins, expected_3bins):
-    X = np.array([0, 1, 2, 3, 9, 10]).reshape(-1, 1)
+    X = np.array([0, 0.5, 2, 3, 9, 10]).reshape(-1, 1)
 
     # with 2 bins
     est = KBinsDiscretizer(n_bins=2, strategy=strategy, encode='ordinal')