fd633602af08b199d87d222a8cfb85acea17b55a,optuna/visualization.py,,_get_intermediate_plot,#Any#,52

Before Change

        return go.Figure(data=[], layout=layout)

    target_state = [TrialState.PRUNED, TrialState.COMPLETE, TrialState.RUNNING]
    dst_df = df[df["state"].isin(target_state)]
    dst_df = dst_df[dst_df["intermediate_values"].isnull().all(axis=1) == False]
    traces = []
    for __, row in dst_df.iterrows():
        trace = go.Scatter(
            x=row["intermediate_values"].index,

After Change

    target_state = [TrialState.PRUNED, TrialState.COMPLETE, TrialState.RUNNING]
    trials = [trial for trial in trials if trial.state in target_state]
    traces = []
    for trial in trials:
        trace = go.Scatter(
            x=tuple(trial.intermediate_values.keys()),
            y=tuple(trial.intermediate_values.values()),
            mode="lines+markers",
            marker={
                "maxdisplayed": 10
            },
            name="Trial{}".format(trial.number)
        )
        traces.append(trace)

    figure = go.Figure(data=traces, layout=layout)

    return figure

Instances