Как вы загружаете модель простых трансформаторов с контрольной точки?

Question

Я использую простые трансформеры (https://github.com/ThilinaRajapakse/simpletransformers) для рабочего проекта. После обучения модели я получаю кучу файлов, выгруженных в папку вывода. Мне было интересно, если бы вы хотели загрузить ранее обученную модель (которая, я полагаю, сохранена в выходной папке), чтобы вы могли сразу запустить ее с новыми данными, как бы это сделать? Изображение файлов в выходной папке

С помощью scikit-learn вы можете легко экспортировать модели в json и загружать их обратно, я подумал, что для этого может быть аналогичный процесс?

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)