explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-11 17:10:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

bugfix tok2vec

Browse Source
This commit is contained in:
svlandeg 2020-06-18 15:24:40 +02:00
parent 01f9ae774c
commit 64fc840a5d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/tok2vec.py
View File

@ -59,10 +59,10 @@ class Tok2Vec(Pipe):
YIELDS (iterator): A sequence of `Doc` objects, in order of input. YIELDS (iterator): A sequence of `Doc` objects, in order of input.
""" """
for docs in minibatch(stream, batch_size): for docs in minibatch(stream, batch_size):
batch = list(batch) docs = list(docs)
tokvecses = self.predict(docs) tokvecses = self.predict(docs)
self.set_annotations(docs, tokvecses) self.set_annotations(docs, tokvecses)
yield from batch yield from docs
def predict(self, docs): def predict(self, docs):
"""Return a single tensor for a batch of documents. """Return a single tensor for a batch of documents.