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

Fix GoldParse argument to tagger.update

Browse Source
This commit is contained in:
Matthew Honnibal 2016-10-16 17:05:09 +02:00
parent 274a4d4272
commit 195d998a12
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/tagger.pyx
View File

@ -211,6 +211,8 @@ cdef class Tagger:
def update(self, Doc tokens, object gold): def update(self, Doc tokens, object gold):
if hasattr(gold, 'tags'): if hasattr(gold, 'tags'):
gold_tag_strs = list(gold.tags) gold_tag_strs = list(gold.tags)
else:
gold_tag_strs = gold
assert len(tokens) == len(gold_tag_strs) assert len(tokens) == len(gold_tag_strs)
for tag in gold_tag_strs: for tag in gold_tag_strs:
if tag != None and tag not in self.tag_names: if tag != None and tag not in self.tag_names: