如何在 Google Colab 中导入保存的模块？

Question

我在 Google Colab 中使用 Spacy 构建了一个 NER 模块。我使用 nlp.to_disk() 函数将它保存到磁盘。

nlp.to_disk("RCM.model")

该模块保存在文件下。我应该如何导入 RCM 模块以进行测试？

我试过下面的代码，但没有用。

from google.colab import drive
my_module = drive.mount('/content/RCM.model', force_remount=True)

Answer 1

如果您保存了一个模型，您可以使用 spacy.load 加载它。

import spacy

spacy.load("RCM.model") # the argument should be the path to the directory