Skip to content

Commit 51edab8

Browse files
edvardlindelofthomasjpfan
authored andcommitted
DOC clarify StratifiedShuffleSplit supports 2d y (scikit-learn#15456)
1 parent 53b2578 commit 51edab8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/model_selection/_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1713,7 +1713,7 @@ def split(self, X, y, groups=None):
17131713
hence ``np.zeros(n_samples)`` may be used as a placeholder for
17141714
``X`` instead of actual training data.
17151715
1716-
y : array-like, shape (n_samples,)
1716+
y : array-like, shape (n_samples,) or (n_samples, n_labels)
17171717
The target variable for supervised learning problems.
17181718
Stratification is done based on the y labels.
17191719

0 commit comments

Comments
 (0)