DOC add link to stratification UG for train_test_split (#18735)

scikit-learn · Nov 10, 2020 · 35b3195 · 35b3195
1 parent 020479a
commit 35b3195
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/doc/modules/cross_validation.rst b/doc/modules/cross_validation.rst
@@ -518,6 +518,8 @@ Here is a visualization of the cross-validation behavior. Note that
 validation that allows a finer control on the number of iterations and
 the proportion of samples on each side of the train / test split.
 
+.. _stratification:
+
 Cross-validation iterators with stratification based on class labels.
 ---------------------------------------------------------------------
 

diff --git a/sklearn/model_selection/_split.py b/sklearn/model_selection/_split.py
@@ -2120,6 +2120,7 @@ def train_test_split(*arrays,
     stratify : array-like, default=None
         If not None, data is split in a stratified fashion, using this as
         the class labels.
+        Read more in the :ref:`User Guide <stratification>`.
 
     Returns
     -------