diff --git a/spacy/syntax/parser.pyx b/spacy/syntax/parser.pyx index 6d31291e3..b46d42547 100644 --- a/spacy/syntax/parser.pyx +++ b/spacy/syntax/parser.pyx @@ -150,7 +150,7 @@ cdef class StepwiseState: self.doc = doc self.stcls = StateClass.init(doc.c, doc.length) self.parser.moves.initialize_state(self.stcls) - self.eg = Example(self.parser.model.n_classes, CONTEXT_SIZE, + self.eg = Example(self.parser.model.nr_class, CONTEXT_SIZE, self.parser.model.n_feats, self.parser.model.n_feats) def __enter__(self):