plot_contour

optuna.visualization.plot_contour(study, params=None, *, target=None, target_name='Objective Value')

将学习（study）中的参数关系绘制为等高线图。

请注意，如果参数包含缺失值，则不会绘制具有缺失值的试验（trial）。

参数：

• study (Study) – 一个 Study 对象，其试验的目标值将被绘制。

• params (list[str] | None) – 要可视化的参数列表。默认为所有参数。

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

  一个函数，用于指定要显示的值。如果为 None 且 study 用于单目标优化，则绘制目标值。

  注意

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

• target_name (str) – 目标值的名称，显示在颜色条上。

返回：

一个 plotly.graph_objects.Figure 对象。

返回类型：

Figure

注意

当 target 参数不为 None 或 Study 的 direction 为 minimize 时，色图（colormap）会反转。

以下代码片段展示了如何将参数关系绘制为等高线图。

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

fig = optuna.visualization.plot_contour(study, params=["x", "y"])
show(fig)