如何从检查点加载 simpletransformers model？

Question

我正在为一个工作项目使用 simpletransformers ( https://github.com/ThilinaRajapakse/simpletransformers )。 在训练 model 后，我得到一堆文件转储到 output 文件夹中。 我想知道，如果您想加载先前训练过的 model（我假设它已保存到 output 文件夹中），以便您可以立即针对新数据运行它，那么 Z34D1F91FB2E514B8576FAB1A75A 会怎么做？ output文件夹中的文件图片

使用 scikit-learn，您可以轻松地将模型导出到 json，然后将它们重新加载，我想可能有类似的过程？

Answer 1

from simpletransformers.classification import ClassificationModel

model = ClassificationModel("bert", "outputs/best_model")
model.predict(['Enter text to classify'])

Answer 2

model = ClassificationModel(model_name, model_path, num_labels=no_labels)