mirror of
https://github.com/explosion/spaCy.git
synced 2025-02-11 17:10:36 +03:00
Fix eg.doc
This commit is contained in:
parent
79288e7110
commit
455dc0d9e2
|
@ -116,7 +116,7 @@ class Morphologizer(Tagger):
|
||||||
d_scores = scores - to_categorical(correct, n_classes=scores.shape[1])
|
d_scores = scores - to_categorical(correct, n_classes=scores.shape[1])
|
||||||
d_scores *= self.model.ops.asarray(known_labels)
|
d_scores *= self.model.ops.asarray(known_labels)
|
||||||
loss = (d_scores**2).sum()
|
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])
|
d_scores = self.model.ops.unflatten(d_scores, [len(d) for d in docs])
|
||||||
return float(loss), d_scores
|
return float(loss), d_scores
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user