训练好的模型预测问题

yhcc / BARTABSA

142 stars 28 forks source link

Open wangludewdrop opened 3 years ago

wangludewdrop commented 3 years ago

您好，训练好的模型怎么用来预测新的数据呢，新手一枚，捣鼓好几天也没搞定，还请指点，最好出一个专门用来预测新数据的文件，不胜感激！

yhcc commented 2 years ago

fastNLP里面有一个简单的方法来生成预测文件就是修改metric的逻辑，在evaluate()函数中存下每个batch的预测结果（例如append到metric的某个属性上），然后再在get_metric的时候写出到某个文件里，

wangludewdrop commented 2 years ago

好的，谢谢您，问题已经解决了

Meatssauce commented 2 years ago

好的，谢谢您，问题已经解决了

我也是同样的问题。请问可否分享一下解决方案的代码