Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed train_parser examples when model_dir isn't None

This commit is contained in:
kendricktan 2016-10-20 23:40:51 +10:00
parent bfe5cb1244
commit f77b3dc677
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/training/train_parser.py
View File

@ -27,10 +27,10 @@ def train_parser(nlp, train_data, left_labels, right_labels):
def main(model_dir=None): def main(model_dir=None):
if model_dir is not None: if model_dir is not None:
model_dir = pathlb.Path(model_dir) model_dir = pathlib.Path(model_dir)
if not model_dir.exists(): if not model_dir.exists():
model_dir.mkdir() model_dir.mkdir()
assert model_dir.isdir() assert model_dir.is_dir()
nlp = spacy.load('en', tagger=False, parser=False, entity=False, vectors=False) nlp = spacy.load('en', tagger=False, parser=False, entity=False, vectors=False)
@ -62,7 +62,7 @@ def main(model_dir=None):
print(word.text, word.dep_, word.head.text) print(word.text, word.dep_, word.head.text)
if model_dir is not None: if model_dir is not None:
with (model_dir / 'config.json').open('wb') as file_: with (model_dir / 'config.json').open('w') as file_:
json.dump(parser.cfg, file_) json.dump(parser.cfg, file_)
parser.model.dump(str(model_dir / 'model')) parser.model.dump(str(model_dir / 'model'))