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

Question

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

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

Answer 1

_foo

Answer 2

from simpletransformers.classification import ClassificationModel

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