1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
|
"""
plot_timeline
=============
.. autofunction:: optuna.visualization.matplotlib.plot_timeline
The following code snippet shows how to plot the timeline of a study.
"""
import time
import optuna
def objective(trial):
x = trial.suggest_float("x", 0, 1)
time.sleep(x * 0.1)
if x > 0.8:
raise ValueError()
if x > 0.4:
raise optuna.TrialPruned()
return x**2
study = optuna.create_study(direction="minimize")
study.optimize(objective, n_trials=50, n_jobs=2, catch=(ValueError,))
optuna.visualization.matplotlib.plot_timeline(study)
|