Tester能否返回测试集的预测结果

fastnlp / fastNLP

fastNLP: A Modularized and Extensible NLP Framework. Currently still in incubation.

https://gitee.com/fastnlp/fastNLP

Apache License 2.0

3.06k stars 450 forks source link

Closed onebula closed 3 years ago

onebula commented 4 years ago

测试集比较大的情况下，Tester跑完一遍只得到了metrics，为了得到预测结果还需要用Predictor再跑一遍，非常费时间。建议Tester也同时返回预测结果

yhcc commented 4 years ago

主要是预测结果这事儿还挺麻烦的，因为不同的model如何组装预测结果（例如可能涉及到去除padding这样）有点差别，所以我们目前对预测结果这块支持本身也不太好。你说的这个建议，我们之后对predictor再做一些升级后再考虑如何结合一下吧，谢谢～