【问题标题】:SpaCy 'nlp.to_disk' is not saving to diskSpaCy 'nlp.to_disk' 未保存到磁盘
【发布时间】: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


【解决方案1】:

感谢polm23 让我找到正确的方向!

我没有添加已定义目录路径的选项。以下是运行模型并保存模型所需的操作。

python train_ner.py -m=en -o=path/to/output/directory -n=1000

这是我的一个愚蠢的新手错误,但我非常感谢帮助。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2012-10-01
    • 2012-11-02
    • 2013-03-29
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多