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

Fix handling of tuple-valued tags in Tagger

Browse Source
This commit is contained in:
Matthew Honnibal 2018-04-03 01:55:06 +02:00
parent 9c5c940441
commit 6d42e0ad8e
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/pipeline.pyx
View File

@ -503,6 +503,8 @@ class Tagger(Pipe):
elif isinstance(tag, list): # Leave these latent for now.
correct[idx] = guesses[idx]
else:
if isinstance(tag, tuple):
tag = tag[0]
correct[idx] = tag_index[tag]
idx += 1
correct = self.model.ops.xp.array(correct, dtype='i')