diff --git a/spacy/syntax/parser.pyx b/spacy/syntax/parser.pyx index c9e1b0412..80c70c333 100644 --- a/spacy/syntax/parser.pyx +++ b/spacy/syntax/parser.pyx @@ -88,14 +88,30 @@ cdef class Parser: self.parse(stcls, eg.c) tokens.set_parse(stcls._sent) - def partial(self, Doc tokens, initial_actions): + def get_state(self, Doc tokens, initial_actions, continue_for=0): cdef StateClass stcls = StateClass.init(tokens.data, tokens.length) self.moves.initialize_state(stcls) cdef object action_name cdef Transition action + cdef Example eg for action_name in initial_actions: - action = self.moves.lookup_transition(action_name) + try: + action = self.moves.lookup_transition(action_name) + except IndexError: + break action.do(stcls, action.label) + else: + eg = Example(self.model.n_classes, CONTEXT_SIZE, + self.model.n_feats, self.model.n_feats) + while not stcls.is_final() and continue_for != 0: + memset(eg.c.scores, 0, eg.c.nr_class * sizeof(weight_t)) + self.moves.set_valid(eg.c.is_valid, stcls) + fill_context(eg.c.atoms, stcls) + self.model.set_scores(eg.c.scores, eg.c.atoms) + eg.guess = arg_max_if_true(eg.c.scores, eg.c.is_valid, self.model.n_classes) + self.moves.c[eg.guess].do(stcls, self.moves.c[eg.c.guess].label) + if stcls.is_final(): + self.moves.finalize_state(stcls) tokens.set_parse(stcls._sent) return stcls