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

[doc] Update for prediction. (#7648)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-02-15 05:01:55 +08:00
committed by GitHub
parent 0d0abe1845
commit 0da7d872ef
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
doc/parameter.rst
View File

@@ -267,12 +267,12 @@ Additional parameters for Dart Booster (``booster=dart``)
If the booster object is DART type, ``predict()`` will perform dropouts, i.e. only
some of the trees will be evaluated. This will produce incorrect results if ``data`` is
not the training data. To obtain correct results on test sets, set ``ntree_limit`` to
not the training data. To obtain correct results on test sets, set ``iteration_range`` to
a nonzero value, e.g.
.. code-block:: python
preds = bst.predict(dtest, ntree_limit=num_round)
preds = bst.predict(dtest, iteration_range=(0, num_round))
* ``sample_type`` [default= ``uniform``]