-
- Downloads
[MRG+1] feature: add beta-threshold early stopping for decision tree growth (#6954)
* feature: add beta-threshold early stopping for decision tree growth * check if value of beta is greater than or equal to 0 * test if default value of beta is 0 and edit input validation error message * feature: separately validate beta for reg. and clf., and add tests for it * feature: add beta to forest-based ensemble methods * feature: add separate condition to determine that beta is float * feature: add beta to gradient boosting estimators * rename parameter to min_impurity_split, edit input validation and associated tests * chore: fix spacing in forest and force recompilation of grad boosting extension * remove trivial comment in grad boost and add whats new * edit wording in test comment / rebuild * rename constant with the same name as our parameter * edit line length for what's new * remove constant and set min_impurity_split to 1e-7 by default * fix docstrings for new default * fix defaults in gradientboosting and forest classes
Showing
- doc/whats_new.rst 5 additions, 0 deletionsdoc/whats_new.rst
- sklearn/ensemble/forest.py 35 additions, 5 deletionssklearn/ensemble/forest.py
- sklearn/ensemble/gradient_boosting.py 17 additions, 6 deletionssklearn/ensemble/gradient_boosting.py
- sklearn/tree/_tree.pxd 2 additions, 0 deletionssklearn/tree/_tree.pxd
- sklearn/tree/_tree.pyx 11 additions, 5 deletionssklearn/tree/_tree.pyx
- sklearn/tree/tests/test_tree.py 51 additions, 0 deletionssklearn/tree/tests/test_tree.py
- sklearn/tree/tree.py 25 additions, 2 deletionssklearn/tree/tree.py
Loading
Please register or sign in to comment