From 3c53ce924cf8a96947cacaa7d860687ccb046a2e Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Tue, 19 Jan 2021 13:14:34 +0100
Subject: [PATCH] R2 is given with more digits in eval script
---
scripts/conll18_ud_eval.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/scripts/conll18_ud_eval.py b/scripts/conll18_ud_eval.py
index 813188c..2b760bd 100755
--- a/scripts/conll18_ud_eval.py
+++ b/scripts/conll18_ud_eval.py
@@ -760,12 +760,15 @@ def main() :
evaluation[metric][0].aligned_total or (evaluation[metric][0].correct if metric == "Words" else "")
))
else :
- print("{:{}}|{:10.2f} |{:10.2f} |{:10.2f} |{}".format(
+ precision = ("{:10.2f}" if abs(evaluation[metric][0].precision) > 1.0 else "{:10.4f}").format(evaluation[metric][0].precision)
+ recall = ("{:10.2f}" if abs(evaluation[metric][0].recall) > 1.0 else "{:10.4f}").format(evaluation[metric][0].recall)
+ f1 = ("{:10.2f}" if abs(evaluation[metric][0].f1) > 1.0 else "{:10.4f}").format(evaluation[metric][0].f1)
+ print("{:{}}|{} |{} |{} |{}".format(
metric,
maxColNameSize,
- evaluation[metric][0].precision,
- evaluation[metric][0].recall,
- evaluation[metric][0].f1,
+ precision,
+ recall,
+ f1,
"{:10.2f}".format(evaluation[metric][0].aligned_accuracy) if evaluation[metric][0].aligned_accuracy is not None else ""
))
--
GitLab