Fix set labels meta

This commit is contained in:
Matthew Honnibal 2019-09-19 00:56:07 +02:00
parent 9d399fe63a
commit e34b4a38b0
1 changed files with 1 additions and 0 deletions

View File

@ -441,6 +441,7 @@ def train(
}
meta.setdefault("name", "model%d" % i)
meta.setdefault("version", version)
meta["labels"] = nlp.meta["labels"]
meta_loc = output_path / ("model%d" % i) / "meta.json"
srsly.write_json(meta_loc, meta)
util.set_env_log(verbose)