Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert changes to Scorer.score_spans

This commit is contained in:
Adriane Boyd 2020-09-25 08:21:30 +02:00
parent 93d7ff309f
commit 50f20cf722
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
spacy/scorer.py
View File

@ -314,14 +314,9 @@ class Scorer:
gold_spans.add(gold_span)
gold_per_type[span.label_].add((span.label_, span.start, span.end - 1))
pred_per_type = {label: set() for label in labels}
align_x2y = example.alignment.x2y
for pred_span in getter(pred_doc, attr):
indices = align_x2y[pred_span.start : pred_span.end].dataXd.ravel()
if len(indices):
g_span = gold_doc[indices[0] : indices[-1]]
span = (pred_span.label_, indices[0], indices[-1])
pred_spans.add(span)
pred_per_type[pred_span.label_].add(span)
for span in example.get_aligned_spans_x2y(getter(pred_doc, attr)):
pred_spans.add((span.label_, span.start, span.end - 1))
pred_per_type[span.label_].add((span.label_, span.start, span.end - 1))
# Scores per label
for k, v in score_per_type.items():
if k in pred_per_type: