explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-15 02:32:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix NER init_gold_batch

Browse Source
This commit is contained in:
Matthew Honnibal 2020-06-23 23:58:55 +02:00
parent 78e9e15e9e
commit d85d063231
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/syntax/ner.pyx
View File

@ -133,9 +133,8 @@ cdef class BiluoPushDown(TransitionSystem):
examples = [eg for eg in examples if self.has_gold(eg)] examples = [eg for eg in examples if self.has_gold(eg)]
states = self.init_batch([eg.predicted for eg in examples]) states = self.init_batch([eg.predicted for eg in examples])
keeps = [i for i, s in enumerate(states) if not s.is_final()] keeps = [i for i, s in enumerate(states) if not s.is_final()]
states = [states[i] for i in keeps]
examples = [examples[i] for i in keeps]
golds = [BiluoGold(self, states[i], examples[i]) for i in keeps] golds = [BiluoGold(self, states[i], examples[i]) for i in keeps]
states = [states[i] for i in keeps]
n_steps = sum([len(s.queue) for s in states]) n_steps = sum([len(s.queue) for s in states])
return states, golds, n_steps return states, golds, n_steps