dask_ml.xgboost.predict

dask_ml.xgboost.predict(client, model, data)

Distributed prediction with XGBoost

Parameters:
client: dask.distributed.Client
model: xgboost.Booster
data: dask array or dataframe
Returns:
Dask.dataframe or dask.array, depending on the input data type

See also

train

Examples

>>> client = Client('scheduler-address:8786')  # doctest: +SKIP
>>> test_data = dd.read_csv('s3://...')  # doctest: +SKIP
>>> model
<xgboost.core.Booster object at ...>
>>> predictions = predict(client, model, test_data)  # doctest: +SKIP