将 spacy 预训练模型下载添加到 yml 文件

Question

我正在编写一个依赖于 spaCy 及其预训练模型的 python 代码。对于这段代码，我想创建一个 yml 文件来安装所有依赖项。到目前为止，它看起来像这样：

name: my_pkg
channels:
  - conda-forge
  - anaconda
  - defaults
dependencies:
  - pandas
  - numpy
  - spacy

但是，为了让 spacy 正常运行，我还需要下载预训练模型。基于 spacy 网站我应该使用python -m spacy download en_core_web_sm 命令来下载模型，但我不知道如何将此命令添加到 yml 文件中。我想知道是否有人可以帮助我修复 yml，使其自动下载 spacy 预训练模型

Answer 1

您可以通过conda找到可以下载的模型：

conda search *en_core_web_md*

---

# Name                       Version           Build  Channel     
spacy-model-en_core_web_md           2.0.0  py27hdea317b_0  conda-forge
spacy-model-en_core_web_md           2.1.0    pyhdea317b_0  conda-forge
spacy-model-en_core_web_md           2.2.0            py_0  conda-forge
spacy-model-en_core_web_md           2.2.0    pyhdea317b_0  conda-forge
spacy-model-en_core_web_md           2.2.5            py_0  conda-forge
spacy-model-en_core_web_md           2.3.0    pyh9f0ad1d_0  conda-forge  
spacy-model-en_core_web_md           2.3.1    pyh9f0ad1d_0  conda-forge

那么对于你的 yml 文件你可以试试：

name: my_pkg
channels:
  - conda-forge
  - anaconda
  - defaults
dependencies:
  - pandas
  - numpy
  - spacy
  - spacy-model-en_core_web_md==2.3.1