explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-13 01:32:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix example

Browse Source
This commit is contained in:
ines 2017-10-26 15:15:08 +02:00
parent 281f88a59c
commit d425ede7e9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/training/train_ner.py
View File

@ -44,7 +44,7 @@ def main(model=None, output_dir=None, n_iter=100):
print("Created blank 'en' model") print("Created blank 'en' model")
# create the built-in pipeline components and add them to the pipeline # create the built-in pipeline components and add them to the pipeline
# ner.create_pipe works for built-ins that are registered with spaCy! # nlp.create_pipe works for built-ins that are registered with spaCy
if 'ner' not in nlp.pipe_names: if 'ner' not in nlp.pipe_names:
ner = nlp.create_pipe('ner') ner = nlp.create_pipe('ner')
nlp.add_pipe(ner, last=True) nlp.add_pipe(ner, last=True)
@ -86,8 +86,9 @@ def main(model=None, output_dir=None, n_iter=100):
# test the saved model # test the saved model
print("Loading from", output_dir) print("Loading from", output_dir)
nlp2 = spacy.load(output_dir)
for text, _ in TRAIN_DATA: for text, _ in TRAIN_DATA:
doc = nlp(text) doc = nlp2(text)
print('Entities', [(ent.text, ent.label_) for ent in doc.ents]) print('Entities', [(ent.text, ent.label_) for ent in doc.ents])
print('Tokens', [(t.text, t.ent_type_, t.ent_iob) for t in doc]) print('Tokens', [(t.text, t.ent_type_, t.ent_iob) for t in doc])