explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-18 12:12:20 +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)
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