From c1e8ec8e9953e653afde35f4fb6402b957dd3788 Mon Sep 17 00:00:00 2001
From: vvye <ekaiser.hellwege@gmail.com>
Date: Wed, 6 Oct 2021 22:22:12 +0200
Subject: [PATCH] Fix error
---
evaluation.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/evaluation.py b/evaluation.py
index 3d5fd04..30bdb56 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -83,7 +83,8 @@ class ResultLogger:
avg_ar1_f, avg_ar2_f, avg_date_f1 = self.average_scores()
avg_gold_sentences, avg_system_sentences, avg_gold_tokens, avg_system_tokens = self.average_stats()
with open(filename, 'w', encoding='utf-8') as f:
- w = csv.DictWriter(f, ['topic', 'gold_timeline', 'ar1_f', 'ar2_f', 'date_f1'], delimiter=';')
+ w = csv.DictWriter(f, ['topic', 'gold_timeline', 'avg_gold_sentences', 'avg_system_sentences',
+ 'avg_gold_tokens', 'avg_system_tokens', 'ar1_f', 'ar2_f', 'date_f1'], delimiter=';')
w.writeheader()
w.writerows(self.results)
w.writerow({
--
GitLab