dask_ml.xgboost.predict
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
Examples
>>> client = Client('scheduler-address:8786') >>> test_data = dd.read_csv('s3://...') >>> model <xgboost.core.Booster object at ...>
>>> predictions = predict(client, model, test_data)