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: