scikit-learn · jeremiedbb · Mar 17, 2021 · Mar 16, 2021 · Mar 16, 2021 · Mar 16, 2021
diff --git a/doc/developers/utilities.rst b/doc/developers/utilities.rst
@@ -196,6 +196,9 @@ Helper Functions
   to ``n``.  Used in :func:`~sklearn.decomposition.dict_learning` and
   :func:`~sklearn.cluster.k_means`.
 
+- :class:`gen_batches`: generator to create slices containing batch size elements 
+  from 0 to ``n``
+
 - :func:`safe_mask`: Helper function to convert a mask to the format expected
   by the numpy array or scipy sparse matrix on which to use it (sparse
   matrices support integer indices only while numpy arrays support both

diff --git a/doc/modules/classes.rst b/doc/modules/classes.rst
@@ -1595,6 +1595,7 @@ Plotting
    utils.extmath.fast_logdet
    utils.extmath.density
    utils.extmath.weighted_mode
+   utils.gen_batches
    utils.gen_even_slices
    utils.graph.single_source_shortest_path_length
    utils.graph_shortest_path.graph_shortest_path

diff --git a/sklearn/utils/__init__.py b/sklearn/utils/__init__.py
@@ -691,6 +691,10 @@ def gen_batches(n, batch_size, *, min_batch_size=0):
     ------
     slice of batch_size elements
 
+    See Also
+    --------
+    gen_even_slices: Generator to create n_packs slices going up to n.
+
     Examples
     --------
     >>> from sklearn.utils import gen_batches
@@ -740,6 +744,11 @@ def gen_even_slices(n, n_packs, *, n_samples=None):
     ------
     slice
 
+    See Also
+    --------
+    gen_batches: Generator to create slices containing batch_size elements
+        from 0 to n.
+
     Examples
     --------
     >>> from sklearn.utils import gen_even_slices