From 8f78426a39e427570161d9faef8bb0dd5da6f9d6 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 26 Sep 2019 12:26:11 +0200 Subject: [PATCH] Made visualise metrics compatible with older result files Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 331af4a..1c0d72d 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -452,14 +452,23 @@ def _visualise_metrics(visualise_precision_recall: callable, else f"{metrics_file}.bin", "rb") as file: metrics = pickle.load(file) - precision_micro = metrics["cumulative_precisions_micro"] - recall_micro = metrics["cumulative_recalls_micro"] + try: + precision_micro = metrics["cumulative_precisions_micro"] + recall_micro = metrics["cumulative_recalls_micro"] + except KeyError: + precision_micro = metrics["cumulative_precision_micro"] + recall_micro = metrics["cumulative_recall_micro"] + visualise_precision_recall(precision_micro, recall_micro, output_path, f"micro-{threshold}" if use_entropy_threshold else "micro") - - precision_macro = metrics["cumulative_precisions_macro"] - recall_macro = metrics["cumulative_recalls_macro"] + + try: + precision_macro = metrics["cumulative_precisions_macro"] + recall_macro = metrics["cumulative_recalls_macro"] + except KeyError: + precision_macro = metrics["cumulative_precision_macro"] + recall_macro = metrics["cumulative_recall_macro"] visualise_precision_recall(precision_macro, recall_macro, output_path, f"macro-{threshold}" if use_entropy_threshold else "macro")