From 88ac5c6e981fa6b4f07262718100e1956f49d2bb Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Fri, 5 Jun 2015 17:12:06 +0200 Subject: [PATCH] * Send beam_width < 0 to greedy 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 4d4a73df8..5fc0be0f9 100644 --- a/spacy/syntax/parser.pyx +++ b/spacy/syntax/parser.pyx @@ -83,14 +83,14 @@ cdef class Parser: def __call__(self, Tokens tokens): if tokens.length == 0: return 0 - if self.cfg.beam_width == 1: + if self.cfg.beam_width <= 1: self._greedy_parse(tokens) else: self._beam_parse(tokens) def train(self, Tokens tokens, GoldParse gold): self.moves.preprocess_gold(gold) - if self.cfg.beam_width == 1: + if self.cfg.beam_width <= 1: return self._greedy_train(tokens, gold) else: return self._beam_train(tokens, gold)