我正在为一个工作项目使用 simpletransformers ( https://github.com/ThilinaRajapakse/simpletransformers )。训练模型后，我将一堆文件转储到输出文件夹中。我想知道，如果您想加载一个先前训练过的模型（我假设它已保存到输出文件夹中），以便您可以立即针对新数据运行它，那么您将如何做呢？输出文件夹中的文件图片
使用 scikit-learn，您可以轻松地将模型导出为 json，然后将它们重新加载，我认为可能有类似的过程？
问问题
1817 次
2 回答
3
from simpletransformers.classification import ClassificationModel
model = ClassificationModel("bert", "outputs/best_model")
model.predict(['Enter text to classify'])
于 2021-06-07T04:33:07.733 回答
-1
model = ClassificationModel(model_name, model_path, num_labels=no_labels)
于 2020-07-31T09:51:53.043 回答