Fix eg.doc

This commit is contained in:
Matthew Honnibal 2020-06-22 01:03:39 +02:00
parent 79288e7110
commit 455dc0d9e2

View File

@ -116,7 +116,7 @@ class Morphologizer(Tagger):
d_scores = scores - to_categorical(correct, n_classes=scores.shape[1])
d_scores *= self.model.ops.asarray(known_labels)
loss = (d_scores**2).sum()
docs = [eg.doc for eg in examples]
docs = [eg.predicted for eg in examples]
d_scores = self.model.ops.unflatten(d_scores, [len(d) for d in docs])
return float(loss), d_scores