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

Add new example for training new entity types

This commit is contained in:
Matthew Honnibal 2017-04-15 16:11:06 +02:00
parent 4884b2c113
commit c729d72fc6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/training/train_new_entity_type.py
View File

@ -29,8 +29,8 @@ def train_ner(nlp, train_data, output_dir):
doc = nlp.make_doc(raw_text) doc = nlp.make_doc(raw_text)
nlp.tagger(doc) nlp.tagger(doc)
loss = nlp.entity.update(doc, gold) loss = nlp.entity.update(doc, gold)
nlp.end_training()
nlp.save_to_directory(output_dir) nlp.save_to_directory(output_dir)
#nlp.end_training(output_dir)
def main(model_name, output_directory=None): def main(model_name, output_directory=None):