8000 [MRG+1] Add note about the size of default random forest model #6276 … · scikit-learn/scikit-learn@7f084b0 · GitHub
[go: up one dir, main page]

Skip to content

Commit 7f084b0

8000
Browse files
Morikkojnothman
authored andcommitted
[MRG+1] Add note about the size of default random forest model #6276 (#8437)
1 parent fcb1403 commit 7f084b0

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

doc/modules/ensemble.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,13 @@ When using bootstrap sampling the generalization accuracy can be estimated
213213
on the left out or out-of-bag samples. This can be enabled by
214214
setting ``oob_score=True``.
215215

216+
.. note::
217+
218+
The size of the model with the default parameters is :math:`O( M * N * log (N) )`,
219+
where :math:`M` is the number of trees and :math:`N` is the number of samples.
220+
In order to reduce the size of the model, you can change these parameters:
221+
``min_samples_split``, ``min_samples_leaf``, ``max_leaf_nodes`` and ``max_depth``.
222+
216223
Parallelization
217224
---------------
218225

0 commit comments

Comments
 (0)
0