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