Skip to content

Commit 43aa3ef

Browse files
committed
Fix docstring
Signed-off-by: Adam Li <adam2392@gmail.com>
1 parent 13e2913 commit 43aa3ef

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

sklearn/ensemble/_forest.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -817,12 +817,11 @@ def get_leaf_node_samples(self, X):
817817
818818
Returns
819819
-------
820-
leaf_node_samples : a list of array-like of shape
821-
(n_leaf_node_samples, n_outputs)
820+
leaf_node_samples : a list of array-like
822821
Each sample is represented by the indices of the training samples that
823822
reached the leaf node. The ``n_leaf_node_samples`` may vary between
824823
samples, since the number of samples that fall in a leaf node is
825-
variable.
824+
variable. Each array-like has shape (n_leaf_node_samples, n_outputs).
826825
"""
827826
check_is_fitted(self)
828827
# Check data

0 commit comments

Comments
 (0)