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: if self.labels:
label_sample = [ label_sample = [
self.model.ops.alloc2f(len(doc), len(self.labels)) self.model.ops.alloc2f(len(doc), len(self.labels))
for doc in docs for doc in doc_sample
] ]
for y in label_sample: for y in label_sample:
y[:, 0] = 1.0 y[:, 0] = 1.0