Optuna/Visualization/Python/plot_optimization_history.py
import optuna
from optuna import Study
import plotly.graph_objects as go
def plot_optimization_history(study: Study, objective_name: str, objective_index: int) -> go.Figure:
fig = optuna.visualization.plot_optimization_history(
study, target_name=objective_name, target=lambda t: t.values[objective_index]
)
return fig