plot_param_importances

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

绘制超参数重要性。

另请参阅

此函数可视化 optuna.importance.get_param_importances() 的结果。

参数:

• study (Study) – 一个优化的 study。

• evaluator (BaseImportanceEvaluator | None) –

  一个重要性评估器对象，指定基于哪种算法进行重要性评估。默认为 FanovaImportanceEvaluator。

  注意

  虽然 Optuna 中默认的重要性评估器是 FanovaImportanceEvaluator，但 Optuna Dashboard 为了运行时性能会使用一个轻量级的评估器，即 PedAnovaImportanceEvaluator，从而产生不同的结果。

• params (list[str] | None) – 要评估的参数名称列表。如果为 None，则评估所有在所有已完成的 trial 中存在的参数。

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

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

  注意

  如果 study 用于多目标优化，可以使用此参数指定要绘制哪个目标。例如，要仅获取第一个目标的超参数重要性，请为目标参数使用 target=lambda t: t.values[0]。

• target_name (str) – 显示在图例中的目标名称。如果 target 为 None，将使用通过 set_metric_names() 设置的名称，这将覆盖此参数。

返回:

一个 plotly.graph_objects.Figure 对象。

返回类型:

Figure

以下代码片段展示了如何绘制超参数重要性。

import optuna
from plotly.io import show


def objective(trial):
    x = trial.suggest_int("x", 0, 2)
    y = trial.suggest_float("y", -1.0, 1.0)
    z = trial.suggest_float("z", 0.0, 1.5)
    return x**2 + y**3 - z**4


sampler = optuna.samplers.RandomSampler(seed=10)
study = optuna.create_study(sampler=sampler)
study.optimize(objective, n_trials=100)

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