From 56a635e5e69b04855d905753ee58024c58a039a2 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 11 Sep 2019 11:35:18 +0200 Subject: [PATCH] Dealt with different keys in stored metrics files Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 05db393..933b20d 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -576,8 +576,13 @@ def _visualise_all(visualise_precision_recall: callable, f1_scores_micro = [] ose_values_micro = [] for metrics in metrics_micro: - precisions_micro.append(metrics["cumulative_precisions_micro"]) - recalls_micro.append(metrics["cumulative_recalls_micro"]) + try: + precisions_micro.append(metrics["cumulative_precisions_micro"]) + recalls_micro.append(metrics["cumulative_recalls_micro"]) + except KeyError: + precisions_micro.append(metrics["cumulative_precision_micro"]) + recalls_micro.append(metrics["cumulative_recall_micro"]) + f1_scores_micro.append(metrics["f1_scores_micro"]) ose_values_micro.append(metrics["cumulative_open_set_error"]) @@ -597,8 +602,12 @@ def _visualise_all(visualise_precision_recall: callable, f1_scores_macro = [] ose_values_macro = [] for metrics in metrics_macro: - precisions_macro.append(metrics["cumulative_precisions_macro"]) - recalls_macro.append(metrics["cumulative_recalls_macro"]) + try: + precisions_macro.append(metrics["cumulative_precisions_macro"]) + recalls_macro.append(metrics["cumulative_recalls_macro"]) + except KeyError: + precisions_macro.append(metrics["cumulative_precision_macro"]) + recalls_macro.append(metrics["cumulative_recall_macro"]) f1_scores_macro.append(metrics["f1_scores_macro"]) ose_values_macro.append(metrics["cumulative_open_set_error"])