Skip to content
Snippets Groups Projects
Commit c1e8ec8e authored by vvye's avatar vvye
Browse files

Fix error

parent 6c6aa67e
No related branches found
No related tags found
No related merge requests found
......@@ -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({
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment