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

Document num_parallel_tree. (#4022)

Browse Source
This commit is contained in:
Jiaming Yuan 2018-12-25 22:00:58 +08:00 committed by GitHub
parent 85939c6a6e
commit 7735252925
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
doc/parameter.rst
View File

@ -192,6 +192,10 @@ Parameters for Tree Booster
- ``cpu_predictor``: Multicore CPU prediction algorithm. - ``cpu_predictor``: Multicore CPU prediction algorithm.
- ``gpu_predictor``: Prediction using GPU. Default when ``tree_method`` is ``gpu_exact`` or ``gpu_hist``. - ``gpu_predictor``: Prediction using GPU. Default when ``tree_method`` is ``gpu_exact`` or ``gpu_hist``.
* ``num_parallel_tree``, [default=1]
- Number of parallel trees constructed during each iteration. This
option is used to support boosted random forest
Additional parameters for Dart Booster (``booster=dart``) Additional parameters for Dart Booster (``booster=dart``)
========================================================= =========================================================

2
src/gbm/gbtree.cc
View File

@ -52,7 +52,7 @@ struct GBTreeTrainParam : public dmlc::Parameter<GBTreeTrainParam> {
.set_default(1) .set_default(1)
.set_lower_bound(1) .set_lower_bound(1)
.describe("Number of parallel trees constructed during each iteration."\ .describe("Number of parallel trees constructed during each iteration."\
" This option is used to support boosted random forest"); " This option is used to support boosted random forest.");
DMLC_DECLARE_FIELD(updater_seq) DMLC_DECLARE_FIELD(updater_seq)
.set_default("grow_colmaker,prune") .set_default("grow_colmaker,prune")
.describe("Tree updater sequence."); .describe("Tree updater sequence.");