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

Fix handling of unseen labels in tagger

Browse Source
This commit is contained in:
Matthew Honnibal 2018-06-25 22:28:59 +02:00
parent 5b56aad4c2
commit 5a65418c40
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/pipeline.pyx
View File

@ -501,7 +501,7 @@ class Tagger(Pipe):
cdef int idx = 0
correct = numpy.zeros((scores.shape[0],), dtype='i')
guesses = scores.argmax(axis=1)
known_labels = numpy.ones((scores.shape[0],), dtype='f')
known_labels = numpy.ones((scores.shape[0], 1), dtype='f')
for gold in golds:
for tag in gold.tags:
if tag is None: