From 440885d59cc1d4854179d65c456550eae13351c5 Mon Sep 17 00:00:00 2001 From: Guillaume Lemaitre Date: Fri, 24 Sep 2021 15:49:14 +0200 Subject: [PATCH] DOC correct the type of estimator attributes --- sklearn/ensemble/_forest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/ensemble/_forest.py b/sklearn/ensemble/_forest.py index 0f3d93b5a0c4d..31ecfa8ee867f 100644 --- a/sklearn/ensemble/_forest.py +++ b/sklearn/ensemble/_forest.py @@ -2454,11 +2454,11 @@ class RandomTreesEmbedding(BaseForest): Attributes ---------- - base_estimator_ : DecisionTreeClassifier instance + base_estimator_ : :class:`~sklearn.tree.ExtraTreeClassifier` instance The child estimator template used to create the collection of fitted sub-estimators. - estimators_ : list of DecisionTreeClassifier instances + estimators_ : list of :class:`~sklearn.tree.ExtraTreeClassifier` instances The collection of fitted sub-estimators. feature_importances_ : ndarray of shape (n_features,)