dask_ml.xgboost.predict
dask_ml.xgboost.predict¶
- dask_ml.xgboost.predict(client, model, data)¶
使用 XGBoost 进行分布式预测
- 参数
- client: dask.distributed.Client
- model: xgboost.Booster
- data: dask 数组或 dataframe
- 返回值
- Dask.dataframe 或 dask.array，取决于输入数据类型
另请参阅
示例
>>> client = Client('scheduler-address:8786') >>> test_data = dd.read_csv('s3://...') >>> model <xgboost.core.Booster object at ...>
>>> predictions = predict(client, model, test_data)