mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-11 00:32:40 +03:00
Fix bug when docs are empty
This commit is contained in:
parent
c83fccfe2a
commit
a4d2b0c293
|
@ -447,6 +447,9 @@ class Tagger(Pipe):
|
||||||
yield from docs
|
yield from docs
|
||||||
|
|
||||||
def predict(self, docs):
|
def predict(self, docs):
|
||||||
|
if not any(len(doc) for doc in docs):
|
||||||
|
# Handle case where there are no tokens in any 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 = []
|
||||||
|
|
Loading…
Reference in New Issue
Block a user