From 04e7acb13f5757d9f3b9ab3826e878f00fc0459e Mon Sep 17 00:00:00 2001 From: Joaquin Torres Bravo Date: Mon, 13 May 2024 11:18:00 +0200 Subject: [PATCH] method names corrected for plots --- model_selection/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_selection/test_models.py b/model_selection/test_models.py index 120fd52..a582546 100644 --- a/model_selection/test_models.py +++ b/model_selection/test_models.py @@ -277,7 +277,7 @@ if __name__ == "__main__": scores_df.at[model_name, score_name] = score_value # Adjust layout and save/show figure plt.tight_layout() - plt.savefig(f'./test_results/roc_pr_curves/{group}_{method}.svg', format='svg', dpi=500) + plt.savefig(f'./test_results/roc_pr_curves/{group}_{method_names[j]}.svg', format='svg', dpi=500) plt.close(fig) # Store the DataFrame in the dictionary with a unique key for each sheet sheet_name = f"{group}_{method_names[j]}" -- 2.24.1