【发布时间】:2020-10-07 18:13:00
【问题描述】:
我试图弄清楚为什么我的自定义 SpaCy NER 模型没有使用 nlp.to_disk 保存到磁盘。我在我的 python 脚本中使用了这个条件:
# save model to output directory
if output_dir is not None:
output_dir = Path(output_dir)
if not output_dir.exists():
output_dir.mkdir()
nlp.to_disk(output_dir)
print("Saved model to", output_dir)
output_dir 在我的脚本顶部定义为:
@plac.annotations(
model=("Model name. Defaults to blank 'en' model.", "option", "m", str),
output_dir=("Optional output directory", "option", "o", Path),
n_iter=("Number of training iterations", "option", "n", int),
)
模型运行没有任何错误,所有输出都是正确的,但它不会保存到磁盘。我不确定我在这里缺少什么。非常感谢任何帮助。
【问题讨论】:
-
您的输出目录参数是可选的。你确定你提供?是否打印了“保存的模型到”?
-
“模型保存到”不打印。哪个应该是对的?
标签: nlp spacy named-entity-recognition