do
dask-optuna
pypi i dask-optuna
do

dask-optuna

Scale Optuna with Dask

by James Bourbeau

0.0.2 (see all)License:MIT
pypi i dask-optuna
Readme

Dask-Optuna

Tests Documentation Pre-commit

Dask-Optuna helps improve integration between Optuna and Dask by leveraging Optuna's existing distributed optimization capabilities to run optimization trials in parallel on a Dask cluster. It does this by providing a Dask-compatible dask_optuna.DaskStorage storage class which wraps an Optuna storage class (e.g. Optuna's in-memory or sqlite storage) and can be used directly by Optuna. For example:

import optuna
import joblib
import dask.distributed
import dask_optuna

def objective(trial):
    x = trial.suggest_uniform("x", -10, 10)
    return (x - 2) ** 2

with dask.distributed.Client() as client:
    # Create a study using Dask-compatible storage
    storage = dask_optuna.DaskStorage()
    study = optuna.create_study(storage=storage)
    # Optimize in parallel on your Dask cluster
    with joblib.parallel_backend("dask"):
        study.optimize(objective, n_trials=100, n_jobs=-1)
    print(f"best_params = {study.best_params}")

Documentation

See the Dask-Optuna documentation for more information.

License

MIT License

GitHub Stars

26

LAST COMMIT

2yrs ago

MAINTAINERS

1

CONTRIBUTORS

3

OPEN ISSUES

3

OPEN PRs

0
VersionTagPublished
0.0.2
2yrs ago
0.0.1
2yrs ago
0.0.0
2yrs ago
No alternatives found
No tutorials found
Add a tutorial