Predict

[rattlesnakey]

想问下，就是训练好模型以后，怎么用它实现predict来搭起服务呢，有具体的参考code吗？谢谢～

[PingNie1]

有的呀。也可以参考这个：https://huggingface.co/transformers/training.html

trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=small_train_dataset,
    eval_dataset=small_eval_dataset,
    compute_metrics=compute_metrics,
)
trainer.evaluate()

[rattlesnakey]

这个是用来验证的哈，我已经找到了，用的是generate这个接口 可以参考这个页面https://github.com/RussianNLP/russian_paraphrasers/blob/master/russian_paraphrasers/paraphrasers/paraphraser_mt5.py | | hengyuan_blcu | | ***@***.*** | 签名由网易邮箱大师定制 On 09/24/2021 ***@***.***> wrote： 有的呀。也可以参考这个：https://huggingface.co/transformers/training.html trainer = Trainer( model=model, args=training_args, train_dataset=small_train_dataset, eval_dataset=small_eval_dataset, compute_metrics=compute_metrics, ) trainer.evaluate() — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.