Update docstrings

zillow · Sep 1, 2024 · 0aeb321 · 0aeb321
1 parent 07b6006
commit 0aeb321
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py
@@ -220,7 +220,7 @@ def _map_indices_to_leaves(
             Bootstrap indices of training samples.
 
         X_leaves_bootstrap : array-like of shape (n_samples, n_outputs)
-            Leaf nodes indices of the bootstrap training samples.
+            Leaf node indices of the bootstrap training samples.
 
         sample_weight : array-like of shape (n_samples,), default=None
             Sample weights. If None, then samples are equally weighted. Splits
@@ -230,7 +230,8 @@ def _map_indices_to_leaves(
             single class carrying a negative weight in either child node.
 
         leaf_subsample : bool
-            Subsample leaves.
+            Subsample leaf nodes. If True, leaves are randomly sampled to size
+            `max_samples_leaf`.
 
         max_node_count: int
             Maximum number of leaf nodes across all trees.