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') # doctest: +SKIP >>> test_data = dd.read_csv('s3://...') # doctest: +SKIP >>> model <xgboost.core.Booster object at ...>
>>> predictions = predict(client, model, test_data) # doctest: +SKIP