Pass path argument to end_training

This commit is contained in:
Matthew Honnibal 2017-03-09 18:42:40 -06:00
parent cd33b39a04
commit f71eeef9bb
1 changed files with 1 additions and 1 deletions

View File

@ -265,7 +265,7 @@ class Language(object):
self.entity = self.Defaults.create_entity(self) self.entity = self.Defaults.create_entity(self)
self.pipeline = self.Defaults.create_pipeline(self) self.pipeline = self.Defaults.create_pipeline(self)
yield Trainer(self, gold_tuples) yield Trainer(self, gold_tuples)
self.end_training() self.end_training(path=path)
def __init__(self, **overrides): def __init__(self, **overrides):
if 'data_dir' in overrides and 'path' not in overrides: if 'data_dir' in overrides and 'path' not in overrides: