2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Dealt with different keys in stored metrics files

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-09-11 11:35:18 +02:00
parent 9d82b9bd5c
commit 56a635e5e6
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/twomartens/masterthesis/cli.py
View File

@ -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"])