xadrianzetx / optuna-fastcontour

A drop-in replacement for Optuna's contour plot

optuna-fastcontour

A drop-in replacement for optuna.visualization.matplotlib.plot_contour, build mostly because I wanted to check out Cython. Still, this implementation about 35% faster than Optuna's master, which itself is fast enough. Algorithm is an adaptation of the one used by Plotly.

Installation

pip install git+https://github.com/xadrianzetx/optuna-fastcontour.git

Usage

import fastcontour
import optuna

def objective(trial):
    x = trial.suggest_float("x", -100, 100)
    y = trial.suggest_categorical("y", [-1, 0, 1])
    return x ** 2 + y

sampler = optuna.samplers.TPESampler(seed=10)
study = optuna.create_study(sampler=sampler)
study.optimize(objective, n_trials=30)
fastcontour.plot_contour(study, params=["x", "y"])

Test

To run tests locally, first you need to compile extensions:

make build

then, install Python dependencies:

pip install .[test]

and finally, run tests with:

pytest -v