如何从检查点加载 simpletransformers 模型？

Question

我在工作项目中使用 simpletransformers ( https://github.com/ThilinaRajapakse/simpletransformers )。训练模型后，我将一堆文件转储到输出文件夹中。我想知道，如果您想加载之前训练过的模型（我假设将其保存到输出文件夹中）以便您可以立即针对新数据运行它，那么该怎么做呢？输出文件夹中文件的图片

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

Answer 1

from simpletransformers.classification import ClassificationModel

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