explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-18 20:13:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Add warning for models not found in parser

Browse Source
This commit is contained in:
Matthew Honnibal 2015-07-08 20:04:55 +02:00
parent 05b28ec9eb
commit adb868bdad
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spacy/syntax/parser.pyx
View File

@ -72,21 +72,20 @@ cdef class Parser:
def __init__(self, StringStore strings, model_dir, transition_system): def __init__(self, StringStore strings, model_dir, transition_system):
if not os.path.exists(model_dir): if not os.path.exists(model_dir):
print >> sys.stderr, "Warning: No model found at", model_dir print >> sys.stderr, "Warning: No model found at", model_dir
self.cfg = Config(labels={}, features=[])
elif not os.path.isdir(model_dir): elif not os.path.isdir(model_dir):
self.cfg = Config(labels={}, features=[])
print >> sys.stderr, "Warning: model path:", model_dir, "is not a directory" print >> sys.stderr, "Warning: model path:", model_dir, "is not a directory"
else: else:
self.cfg = Config.read(model_dir, 'config') self.cfg = Config.read(model_dir, 'config')
self.moves = transition_system(strings, self.cfg.labels) self.moves = transition_system(strings, self.cfg.labels)
templates = get_templates(self.cfg.features) templates = get_templates(self.cfg.features)
self.model = Model(self.moves.n_moves, templates, model_dir) self.model = Model(self.moves.n_moves, templates, model_dir)
def __call__(self, Doc tokens): def __call__(self, Doc tokens):
if self.cfg.get('beam_width', 0) < 1: if self.model is not None:
self._greedy_parse(tokens) if self.cfg.get('beam_width', 0) < 1:
else: self._greedy_parse(tokens)
self._beam_parse(tokens) else:
self._beam_parse(tokens)
def train(self, Doc tokens, GoldParse gold): def train(self, Doc tokens, GoldParse gold):
self.moves.preprocess_gold(gold) self.moves.preprocess_gold(gold)