From 05b28ec9eb8d34db9f7e4a2222b83f623baecc0d Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Wed, 8 Jul 2015 20:02:13 +0200 Subject: [PATCH] * Add warning for models not found in parser --- spacy/syntax/parser.pyx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/syntax/parser.pyx b/spacy/syntax/parser.pyx index 2427d4a74..6db9ce8a6 100644 --- a/spacy/syntax/parser.pyx +++ b/spacy/syntax/parser.pyx @@ -83,14 +83,14 @@ cdef class Parser: self.model = Model(self.moves.n_moves, templates, model_dir) def __call__(self, Doc tokens): - if self.cfg.get('beam_width', 1) < 1: + if self.cfg.get('beam_width', 0) < 1: self._greedy_parse(tokens) else: self._beam_parse(tokens) def train(self, Doc tokens, GoldParse gold): self.moves.preprocess_gold(gold) - if self.cfg.beam_width < 1: + if self.cfg.get('beam_width', 0) < 1: return self._greedy_train(tokens, gold) else: return self._beam_train(tokens, gold)