adding safe_indexing to _shuffle function

scikit-learn · Dec 28, 2016 · 15a48bf · 15a48bf
1 parent 9ea5c9e
commit 15a48bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sklearn/cross_validation.py b/sklearn/cross_validation.py
@@ -1772,7 +1772,7 @@ def _shuffle(y, labels, random_state):
         for label in np.unique(labels):
             this_mask = (labels == label)
             ind[this_mask] = random_state.permutation(ind[this_mask])
-    return y[ind]
+    return safe_indexing(y, ind)
 
 
 def check_cv(cv, X=None, y=None, classifier=False):

diff --git a/sklearn/model_selection/_validation.py b/sklearn/model_selection/_validation.py
@@ -636,7 +636,7 @@ def _shuffle(y, groups, random_state):
         for group in np.unique(groups):
             this_mask = (groups == group)
             indices[this_mask] = random_state.permutation(indices[this_mask])
-    return y[indices]
+    return safe_indexing(y, indices)
 
 
 def learning_curve(estimator, X, y, groups=None,