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

Made visualise metrics compatible with older result files

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-09-26 12:26:11 +02:00
parent dbe7b5d844
commit 8f78426a39
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -452,14 +452,23 @@ def _visualise_metrics(visualise_precision_recall: callable,
else f"{metrics_file}.bin", "rb") as file: else f"{metrics_file}.bin", "rb") as file:
metrics = pickle.load(file) metrics = pickle.load(file)
try:
precision_micro = metrics["cumulative_precisions_micro"] precision_micro = metrics["cumulative_precisions_micro"]
recall_micro = metrics["cumulative_recalls_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, visualise_precision_recall(precision_micro, recall_micro,
output_path, f"micro-{threshold}" output_path, f"micro-{threshold}"
if use_entropy_threshold else "micro") if use_entropy_threshold else "micro")
try:
precision_macro = metrics["cumulative_precisions_macro"] precision_macro = metrics["cumulative_precisions_macro"]
recall_macro = metrics["cumulative_recalls_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, visualise_precision_recall(precision_macro, recall_macro,
output_path, f"macro-{threshold}" output_path, f"macro-{threshold}"
if use_entropy_threshold else "macro") if use_entropy_threshold else "macro")