plot_optimization_history

optuna.visualization.plot_optimization_history(study, *, target=None, target_name='Objective Value', error_bar=False)[源代码]

绘制研究中所有试验的优化历史记录。

参数:
  • study (Study | Sequence[Study]) – 一个 Study 对象,其试验根据目标值进行绘制。如果你想比较这些优化历史,可以传递多个研究对象。

  • target (Callable[[FrozenTrial], float] | None) –

    A function to specify the value to display. If it is None and study is being used for single-objective optimization, the objective values are plotted.

    备注

    如果 study 用于多目标优化,请指定此参数。

  • target_name (str) – 在轴标签和图例上显示的目标名称。

  • error_bar (bool) – 一个显示误差线的标志。

返回:

一个 plotly.graph_objects.Figure 对象。

返回类型:

Figure

以下代码片段展示了如何绘制优化历史。

import optuna
from plotly.io import show


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=10)

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

脚本总运行时间: (0 分钟 0.048 秒)

由 Sphinx-Gallery 生成的图库