explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-18 20:22:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix score overwriting bug

Browse Source
This commit is contained in:
Kádár Ákos 2022-03-30 17:28:20 +02:00
parent 7ff99a3acc
commit 63a41ba50a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/pipeline/coref.py
View File

@ -636,14 +636,14 @@ class SpanPredictor(TrainablePipe):
evaluator.update(cluster_info) evaluator.update(cluster_info)
score = { score = {
"coref_f": evaluator.get_f1(), "coref_span_f": evaluator.get_f1(),
"coref_p": evaluator.get_precision(), "coref_span_p": evaluator.get_precision(),
"coref_r": evaluator.get_recall(), "coref_span_r": evaluator.get_recall(),
} }
scores.append(score) scores.append(score)
out = {} out = {}
for field in ("f", "p", "r"): for field in ("f", "p", "r"):
fname = f"coref_{field}" fname = f"coref_span_{field}"
out[fname] = mean([ss[fname] for ss in scores]) out[fname] = mean([ss[fname] for ss in scores])
return out return out