Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix document for tree methods. (#6633)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-01-25 15:52:08 +08:00 committed by GitHub
parent fec66d033a
commit 561809200a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
doc/parameter.rst
View File

@ -195,7 +195,7 @@ Parameters for Tree Booster
* ``grow_policy`` [default= ``depthwise``]
- Controls a way new nodes are added to the tree.
- Currently supported only if ``tree_method`` is set to ``hist``.
- Currently supported only if ``tree_method`` is set to ``hist`` or ``gpu_hist``.
- Choices: ``depthwise``, ``lossguide``
- ``depthwise``: split at nodes closest to the root.
@ -207,7 +207,7 @@ Parameters for Tree Booster
* ``max_bin``, [default=256]
- Only used if ``tree_method`` is set to ``hist``.
- Only used if ``tree_method`` is set to ``hist`` or ``gpu_hist``.
- Maximum number of discrete bins to bucket continuous features.
- Increasing this number improves the optimality of splits at the cost of higher computation time.

3
doc/treemethod.rst
View File

@ -1,8 +1,7 @@
####################
XGBoost Tree methods
XGBoost Tree Methods
####################
For training boosted tree models, there are 2 parameters used for choosing algorithms,
namely ``updater`` and ``tree_method``. XGBoost has 4 builtin tree methods, namely
``exact``, ``approx``, ``hist`` and ``gpu_hist``. Along with these tree methods, there