如何从检查点加载 simpletransformers model？

Question

I'm using simpletransformers ( https://github.com/ThilinaRajapakse/simpletransformers ) for a work project. After training a model, I get a bunch of files dumped to the output folder. I was wondering, if you wanted to load in a previously trained model (which I assume is saved to the output folder) so that you can immediately run it against new data, how would one go about doing this? Picture of files in output folder

With scikit-learn, you can export models to json easily, and load them back in, I figured there might be a similar process for this?

Answer 1

from simpletransformers.classification import ClassificationModel

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

Answer 2

model = ClassificationModel(model_name, model_path, num_labels=no_labels)