explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-26 09:14:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix labeller loss for unseen labels

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-22 10:41:20 -05:00
parent c9760b2104
commit a7ee63c0ac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline.pyx
View File

@ -283,7 +283,7 @@ class NeuralLabeller(NeuralTagger):
guesses = scores.argmax(axis=1) guesses = scores.argmax(axis=1)
for gold in golds: for gold in golds:
for tag in gold.labels: for tag in gold.labels:
if tag is None: if tag is None or tag not in self.labels:
correct[idx] = guesses[idx] correct[idx] = guesses[idx]
else: else:
correct[idx] = self.labels[tag] correct[idx] = self.labels[tag]