Skip to content

Commit 5ea317b

Browse files
authored
Merge pull request scikit-learn#6944 from proinsias/proinsias-rfc-doc-defaults
DOC Add missing parameter defaults for some ensembles
2 parents e2a2b4d + 0e59633 commit 5ea317b

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

sklearn/ensemble/forest.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -808,7 +808,7 @@ class RandomForestClassifier(ForestClassifier):
808808
bootstrap : boolean, optional (default=True)
809809
Whether bootstrap samples are used when building trees.
810810
811-
oob_score : bool
811+
oob_score : bool (default=False)
812812
Whether to use out-of-bag samples to estimate
813813
the generalization accuracy.
814814
@@ -830,7 +830,8 @@ class RandomForestClassifier(ForestClassifier):
830830
and add more estimators to the ensemble, otherwise, just fit a whole
831831
new forest.
832832
833-
class_weight : dict, list of dicts, "balanced", "balanced_subsample" or None, optional
833+
class_weight : dict, list of dicts, "balanced",
834+
"balanced_subsample" or None, optional (default=None)
834835
Weights associated with classes in the form ``{class_label: weight}``.
835836
If not given, all classes are supposed to have weight one. For
836837
multi-output problems, a list of dicts can be provided in the same
@@ -1182,7 +1183,7 @@ class ExtraTreesClassifier(ForestClassifier):
11821183
and add more estimators to the ensemble, otherwise, just fit a whole
11831184
new forest.
11841185
1185-
class_weight : dict, list of dicts, "balanced", "balanced_subsample" or None, optional
1186+
class_weight : dict, list of dicts, "balanced", "balanced_subsample" or None, optional (default=None)
11861187
Weights associated with classes in the form ``{class_label: weight}``.
11871188
If not given, all classes are supposed to have weight one. For
11881189
multi-output problems, a list of dicts can be provided in the same
@@ -1460,10 +1461,10 @@ class RandomTreesEmbedding(BaseForest):
14601461
14611462
Parameters
14621463
----------
1463-
n_estimators : int
1464+
n_estimators : integer, optional (default=10)
14641465
Number of trees in the forest.
14651466
1466-
max_depth : int
1467+
max_depth : integer, optional (default=5)
14671468
The maximum depth of each tree. If None, then nodes are expanded until
14681469
all leaves are pure or until all leaves contain less than
14691470
min_samples_split samples.

0 commit comments

Comments
 (0)