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

Add DaskDeviceQuantileDMatrix demo. (#6156)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-09-24 14:08:28 +08:00
committed by GitHub
parent 678ea40b24
commit 78d72ef936
3 changed files with 63 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python-package/xgboost/dask.py
View File

@@ -854,7 +854,8 @@ def predict(client, model, data, missing=numpy.nan, **kwargs):
model: A Booster or a dictionary returned by `xgboost.dask.train`.
The trained model.
data: DaskDMatrix/dask.dataframe.DataFrame/dask.array.Array
Input data used for prediction.
Input data used for prediction. When input is a dataframe object,
prediction output is a series.
missing: float
Used when input data is not DaskDMatrix. Specify the value
considered as missing.