From 63a41ba50abd16c8b945bb39d8beff2879031cc2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?K=C3=A1d=C3=A1r=20=C3=81kos?= Date: Wed, 30 Mar 2022 17:28:20 +0200 Subject: [PATCH] fix score overwriting bug --- spacy/pipeline/coref.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spacy/pipeline/coref.py b/spacy/pipeline/coref.py index f0862c844..25a353405 100644 --- a/spacy/pipeline/coref.py +++ b/spacy/pipeline/coref.py @@ -636,14 +636,14 @@ class SpanPredictor(TrainablePipe): evaluator.update(cluster_info) score = { - "coref_f": evaluator.get_f1(), - "coref_p": evaluator.get_precision(), - "coref_r": evaluator.get_recall(), + "coref_span_f": evaluator.get_f1(), + "coref_span_p": evaluator.get_precision(), + "coref_span_r": evaluator.get_recall(), } scores.append(score) out = {} for field in ("f", "p", "r"): - fname = f"coref_{field}" + fname = f"coref_span_{field}" out[fname] = mean([ss[fname] for ss in scores]) return out