plot_timeline

optuna.visualization.plot_timeline(study)

绘制研究的时间线图。

参数:

study (Study) – 一个 Study 对象，其试验（trials）的生命周期将被绘制出来。

返回值:

一个 plotly.graph_objects.Figure 对象。

返回类型:

Figure

以下代码片段展示了如何绘制研究的时间线图。时间线图可以可视化执行时间重叠的试验（例如，在分布式环境中）。

import time

import optuna
from plotly.io import show


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,)
)

fig = optuna.visualization.plot_timeline(study)
show(fig)