From 5df06bbfa63c807d6911c7b6fb93483fe6f666ae Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 3 Sep 2019 12:44:35 +0200 Subject: [PATCH] Fixed serialisation of scores Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index e0da7fe..dc50b55 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -485,11 +485,11 @@ def _visualise_metrics(visualise_precision_recall: callable, "max_f1_score_micro": max_f1_score_micro, "precision_at_max_f1_micro": precision_at_max_f1_micro, "recall_at_max_f1_micro": recall_at_max_f1_micro, - "ose_at_max_f1_micro": ose_at_max_f1_micro, + "ose_at_max_f1_micro": int(ose_at_max_f1_micro), "max_f1_score_macro": max_f1_score_macro, "precision_at_max_f1_macro": precision_at_max_f1_macro, "recall_at_max_f1_macro": recall_at_max_f1_macro, - "ose_at_max_f1_macro": ose_at_max_f1_macro, + "ose_at_max_f1_macro": int(ose_at_max_f1_macro), }, file)