b104ef8ea5f6e98b0b05a5cf068bba0c8689d445,rankeval/visualization/effectiveness.py,,plot_tree_wise_average_contribution,#,462

Before Change

            axes.set_xlabel("Number of trees")
            axes.legend(performance.coords["model"].values)

            plt.tight_layout()


def plot_query_wise_performance(performance, compare="model"):
    for dataset in performance.coords["dataset"].values:

After Change

        
        for i, model in enumerate(performance.coords["model"].values):
            k_values = performance.sel(dataset=dataset, model=model)
            axes[i, ].plot(k_values.values)
            axes[i, 0].legend((model,), loc="upper center")

        axes[i, 0].set_xlabel("Number of trees")

Instances