plot_slice

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

使用 Matplotlib 绘制研究中的参数关系切片图。

另请参阅

请参阅 optuna.visualization.plot_slice() 以获取示例。

参数:
  • study (Study) – 一个 Study 对象,其试验(trials)的目标值将被绘制。

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

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

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

    注意

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

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

返回值:

一个 matplotlib.axes.Axes 对象。

返回类型:

Axes

注意

在 v2.2.0 中作为实验性功能添加。接口在后续版本中可能会在未提前通知的情况下更改。请参阅 https://github.com/optuna/optuna/releases/tag/v2.2.0

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

Slice Plot
/home/docs/checkouts/readthedocs.org/user_builds/optuna/checkouts/stable/docs/visualization_matplotlib_examples/optuna.visualization.matplotlib.slice.py:25: ExperimentalWarning:

plot_slice is experimental (supported from v2.2.0). The interface can change in the future.


array([<Axes: xlabel='x', ylabel='Objective Value'>, <Axes: xlabel='y'>],
      dtype=object)

import optuna


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)

optuna.visualization.matplotlib.plot_slice(study, params=["x", "y"])

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

由 Sphinx-Gallery 生成的图库