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

Ensure predict leaf output 1-dim vector where there's only 1 tree. (#6889)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-04-23 15:07:48 +08:00
committed by GitHub
parent 54afa3ac7a
commit 8760ec4827
3 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
doc/prediction.rst
View File

@@ -54,8 +54,8 @@ After 1.4 release, we added a new parameter called ``strict_shape``, one can set
Output is a 4-dim array with ``(n_samples, n_iterations, n_classes, n_trees_in_forest)``
as shape. ``n_trees_in_forest`` is specified by the ``numb_parallel_tree`` during
training. When strict shape is set to False, output is a 2-dim array with last 3 dims
concatenated into 1. When using ``apply`` method in scikit learn interface, this is set
to False by default.
concatenated into 1. Also the last dimension is dropped if it eqauls to 1. When using
``apply`` method in scikit learn interface, this is set to False by default.
Other than these prediction types, there's also a parameter called ``iteration_range``,