plot_contour

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

スタディのパラメータ間の関係を等高線プロットとして表示します。

注意: パラメータに欠損値が含まれている場合、欠損値のあるトライアルはプロットされません。

Parameters:

• study (Study) – ターゲット値をプロットする対象の Study オブジェクト

• params (list[str] | None) – 可視化するパラメータのリスト。デフォルトは全てのパラメータ

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

  表示する値を指定する関数。None で study が単一目的最適化の場合、目的値がプロットされます

  Note

  多目的最適化の場合はこの引数を指定する必要があります

• target_name (str) – カラーバーに表示するターゲットの名称

Returns:

plotly.graph_objects.Figure オブジェクト

Return type:

Figure

Note

target 引数が None でない場合、または Study の direction が minimize の場合、カラーマップが反転されます

The following code snippet shows how to plot the parameter relationship as contour plot.

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)