Machine Learning#

Dask is tightly integrated with common Python machine learning libraries like Scikit-Learn, XGBoost, LightGBM, Optuna, and PyTorch. This makes it easy to train models at scale, and run predictions on large scale data.

import xgboost.dask
import optuna.integrations.dask

def objective(trial):
    hyperparameters = {
    xgboost.dask.train(..., **hyperparameters)

study.optimize(objective, ...)

For more in-depth machine learning examples consider the following: