加载自定义训练的 spaCy 模型

Question

我正在尝试加载我之前训练过的 spaCy 文本分类模型。训练完成后，模型被保存到en_textcat_demo-0.0.0.tar.gz文件中。

我想在 jupyter notebook 中使用这个模型，但是当我这样做时

import spacy
spacy.load("spacy_files/en_textcat_demo-0.0.0.tar.gz")

我明白了

OSError: [E053] Could not read meta.json from spacy_files/en_textcat_demo-0.0.0.tar.gz

在此处加载我的模型的正确方法是什么？

Answer 1

您需要解压缩 tar.gz 文件或使用 pip 安装它。

如果解压它，会生成一个目录，您可以将目录名称作为参数提供给 spaCy 加载。

如果您使用 pip install，它将与您的其他库放在一起，您可以像使用预训练的 spaCy 模型一样使用模型名称。