diff --git a/sklearn_questions.py b/sklearn_questions.py index 539ad9f..fa701a4 100644 --- a/sklearn_questions.py +++ b/sklearn_questions.py @@ -176,6 +176,7 @@ def get_n_splits(self, X, y=None, groups=None): def split(self, X, y, groups=None): """Generate indices to split data into training and test set. + Parameters ---------- X : array-like of shape (n_samples, n_features)