explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-11 12:18:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of ssh://github.com/explosion/spaCy

Browse Source
This commit is contained in:
Matthew Honnibal 2016-10-20 14:43:09 +02:00
commit 212b68de81
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_ner.py
View File

@ -22,10 +22,10 @@ def train_ner(nlp, train_data, entity_types):
def main(model_dir=None):
if model_dir is not None:
model_dir = pathlb.Path(model_dir)
model_dir = pathlib.Path(model_dir)
if not model_dir.exists():
model_dir.mkdir()
assert model_dir.isdir()
assert model_dir.is_dir()
nlp = spacy.load('en', parser=False, entity=False, vectors=False)
@ -49,7 +49,7 @@ def main(model_dir=None):
print(word.text, word.tag_, word.ent_type_, word.ent_iob)
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(ner.cfg, file_)
ner.model.dump(str(model_dir / 'model'))