e3d1e6dfd06c35dec767c53b930fa69fc2ff7b12,ludwig/utils/visualization_utils.py,,hyperopt_pair_plot,#Any#Any#Any#,1292

Before Change

    params = list(hyperopt_results_df.keys())
    params.remove(score_name)
    num_param = len(params)
    pairs_params = {(param1, param2) for param1 in params for param2 in params}

    fig = plt.figure()
    fig.subplots_adjust(hspace=0.4, wspace=0.4)
    for i, param_pair in enumerate(pairs_params):

After Change

    gs = fig.add_gridspec(num_param, num_param)

    for i, param1 in enumerate(params):
        for j, param2 in enumerate(params):
            if i != j:
                ax = fig.add_subplot(gs[i, j])
                heatmap = hyperopt_results_df.pivot_table(
                    index=param1,
                    columns=param2,
                    values=score_name,
                    aggfunc="mean"
                )
                sns.heatmap(
                    heatmap,
                    linewidths=1,
                    cmap="viridis",
                    ax=ax,
                )

    plt.tight_layout(pad=5)

    if filename:
        plt.savefig(filename)

Instances