This commit is contained in:
Matthew Honnibal 2020-08-28 15:13:48 +02:00
parent 237cfa2053
commit 472eb28716

View File

@ -289,7 +289,7 @@ class Tagger(Pipe):
if self.labels:
label_sample = [
self.model.ops.alloc2f(len(doc), len(self.labels))
for doc in docs
for doc in doc_sample
]
for y in label_sample:
y[:, 0] = 1.0