plot_parallel_coordinate

optuna.visualization.plot_parallel_coordinate(study, params=None, *, target=None, target_name='Objective Value')[source]

绘制研究中高维参数之间的关系图。

请注意,如果参数包含缺失值,则不会绘制带有缺失值的试验。

参数:
  • study (Study) – 一个 Study 对象,其试验将根据目标值进行绘制。

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

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

    用于指定要显示的值的函数。如果为 Nonestudy 用于单目标优化,则绘制目标值。

    注意

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

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

返回值:

一个 plotly.graph_objects.Figure 对象。

返回类型:

Figure

注意

target 参数不为 NoneStudydirectionminimize 时,色谱图将反转。

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

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_parallel_coordinate(study, params=["x", "y"])
show(fig)

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

画廊由 Sphinx-Gallery 生成