Python 如何从检查点加载simpletransformers模型?

Python 如何从检查点加载simpletransformers模型?,python,model,loading,simpletransformers,Python,Model,Loading,Simpletransformers,我正在为一个工作项目使用simpletransformers()。在训练了一个模型之后,我得到了一堆文件,这些文件被转储到输出文件夹中。我想知道,如果你想加载一个以前训练过的模型(我假设它保存在输出文件夹中),这样你就可以立即对新数据运行它,你会怎么做呢 使用scikit learn,您可以轻松地将模型导出为json,并将其加载回,我想可能会有类似的过程?请不要只发布代码作为答案,还要解释代码的作用以及它如何解决问题。带有解释的答案通常更有帮助,质量更好,更容易吸引选票。 model = Cla

我正在为一个工作项目使用simpletransformers()。在训练了一个模型之后,我得到了一堆文件,这些文件被转储到输出文件夹中。我想知道,如果你想加载一个以前训练过的模型(我假设它保存在输出文件夹中),这样你就可以立即对新数据运行它,你会怎么做呢


使用scikit learn,您可以轻松地将模型导出为json,并将其加载回,我想可能会有类似的过程?

请不要只发布代码作为答案,还要解释代码的作用以及它如何解决问题。带有解释的答案通常更有帮助,质量更好,更容易吸引选票。
model = ClassificationModel(model_name, model_path, num_labels=no_labels)
from simpletransformers.classification import ClassificationModel

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