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

* Fix NER scoring

This commit is contained in:
Matthew Honnibal 2015-03-26 03:20:00 +01:00
parent 1320bd19db
commit f4cc222ec3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/scorer.py
View File

@ -51,7 +51,7 @@ class Scorer(object):
self.tags_corr += token.tag_ == gold.tags[i] self.tags_corr += token.tag_ == gold.tags[i]
self.n_tokens += 1 self.n_tokens += 1
gold_ents = set((start, end, label) for (start, end, label) in gold.ents) gold_ents = set((start, end, label) for (start, end, label) in gold.ents)
guess_ents = set(tokens.ents) guess_ents = set((e.start, e.end, e.label_) for e in tokens.ents)
if verbose and gold_ents: if verbose and gold_ents:
for start, end, label in guess_ents: for start, end, label in guess_ents:
mark = 'T' if (start, end, label) in gold_ents else 'F' mark = 'T' if (start, end, label) in gold_ents else 'F'