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:

• XGBoost
• Hyperparameter Optimization
• XGBoost and Optuna
• GPU PyTorch and Optuna

previous

Visualize 1,000,000,000 Points

next

XGBoost