Fix training

This commit is contained in:
Matthew Honnibal 2017-04-16 15:28:27 -05:00
parent 17c9fffb9e
commit 4efd6fb9d6

View File

@ -239,8 +239,7 @@ class Language(object):
self.pipeline = self.Defaults.create_pipeline(self)
yield Trainer(self, gold_tuples)
self.end_training()
self.save_to_directory(path, deps=self.parser.cfg, ner=self.entity.cfg,
pos=self.tagger.cfg)
self.save_to_directory(path)
def __init__(self, **overrides):
if 'data_dir' in overrides and 'path' not in overrides: