Skip to content

Commit 2d4de9a

Browse files
committed
Fix the predict quantiles docstring
Signed-off-by: Adam Li <adam2392@gmail.com>
1 parent fe3072f commit 2d4de9a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sklearn/ensemble/_forest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -733,9 +733,9 @@ def predict_quantiles(self, X, quantiles=0.5, method="nearest"):
733733
734734
Returns
735735
-------
736-
y : ndarray of shape (n_samples, n_quantiles) or
737-
(n_samples, n_quantiles, n_outputs)
738-
The predicted values.
736+
y : ndarray of shape (n_samples, n_quantiles, [n_output])
737+
The predicted values. The ``n_outputs`` dimension is present only
738+
for multi-output regressors.
739739
"""
740740
if not self.store_leaf_values:
741741
raise RuntimeError(

0 commit comments

Comments
 (0)