python - 如何从检查点加载简单的变形金刚模型？

Question

我正在为一个工作项目使用 simpletransformers ( https://github.com/ThilinaRajapakse/simpletransformers )。训练模型后，我将一堆文件转储到输出文件夹。我想知道，如果你想加载一个以前训练过的模型(我假设它保存在输出文件夹中)以便你可以立即针对新数据运行它，你会怎么做呢？ Picture of files in output folder

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

Answer 1

from simpletransformers.classification import ClassificationModel

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