如何从检查点加载simpletransformers模型？
如何从检查点加载simpletransformers模型？
提问于 2020-06-09 06:00:49
我正在为一个工作项目使用simpletransformers (https://github.com/ThilinaRajapakse/simpletransformers)。训练完模型后，我得到一堆文件，它们被转储到输出文件夹中。我想知道，如果你想加载一个以前训练过的模型(我假设它保存在输出文件夹中)，这样你就可以立即对新数据运行它，你会怎么做呢？Picture of files in output folder
使用scikit-learn，您可以轻松地将模型导出到json，并将其加载回来，我想这可能会有类似的过程？
回答 2
Stack Overflow用户
发布于 2021-06-07 12:33:08
from simpletransformers.classification import ClassificationModel
model = ClassificationModel("bert", "outputs/best_model")
model.predict(['Enter text to classify'])Stack Overflow用户
发布于 2020-07-31 17:51:53
model = ClassificationModel(model_name, model_path, num_labels=no_labels)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/62271872
复制相关文章
相似问题
腾讯云开发者
