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

Don't collate model unless training succeeds

Browse Source
This commit is contained in:
Matthew Honnibal 2018-06-25 16:36:42 +02:00
parent 24dfbb8a28
commit c4698f5712
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spacy/cli/train.py
View File

@ -187,14 +187,15 @@ def train(lang, output_dir, train_data, dev_data, n_iter=30, n_sents=0,
with nlp.use_params(optimizer.averages): with nlp.use_params(optimizer.averages):
final_model_path = output_path / 'model-final' final_model_path = output_path / 'model-final'
nlp.to_disk(final_model_path) nlp.to_disk(final_model_path)
components = [] components = []
if not no_parser: if not no_parser:
components.append('parser') components.append('parser')
if not no_tagger: if not no_tagger:
components.append('tagger') components.append('tagger')
if not no_entities: if not no_entities:
components.append('ner') components.append('ner')
_collate_best_model(meta, output_path, components) _collate_best_model(meta, output_path, components)
def _collate_best_model(meta, output_path, components): def _collate_best_model(meta, output_path, components):
bests = {} bests = {}