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

Fix bug when docs are empty

This commit is contained in:
Matthew Honnibal 2018-06-29 13:56:29 +02:00
parent df3668da12
commit e0860bcfb3
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

@ -449,7 +449,7 @@ class Tagger(Pipe):
def predict(self, docs): def predict(self, docs):
if not any(len(doc) for doc in docs): if not any(len(doc) for doc in docs):
# Handle case where there are no tokens in any docs. # Handle case where there are no tokens in any docs.
return [self.model.ops.allocate((0, self.model.nO) for doc in docs] return [self.model.ops.allocate((0, self.model.nO)) for doc in docs]
tokvecs = self.model.tok2vec(docs) tokvecs = self.model.tok2vec(docs)
scores = self.model.softmax(tokvecs) scores = self.model.softmax(tokvecs)
guesses = [] guesses = []