如何使用训练好的模型进行预测？

649453932 / Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。

MIT License

5.19k stars 1.22k forks source link

Open leoxu1007 opened 2 years ago

leoxu1007 commented 2 years ago

这个程序貌似只能用来训练，不能够用来预测

caichangzhao commented 2 years ago

把evaluate改一下就行了

muzimua commented 2 years ago

把evaluate改一下就行了你好，我也想知道怎么用训练好的模型进行预测，请问具体要把evaluate怎样进行修改呢？

caichangzhao commented 2 years ago

把evaluate改一下就行了你好，我也想知道怎么用训练好的模型进行预测，请问具体要把evaluate怎样进行修改呢？

就是按照训练时候的传入格式传入预测数据，然后跟test中一样调用模型输出结果就行

cingtiye commented 2 years ago

这个程序貌似只能用来训练，不能够用来预测

leoxu1007 commented 2 years ago

把evaluate改一下就行了

你好，请问具体一点要怎样改呢？

cingtiye commented 2 years ago

把evaluate改一下就行了

你好，请问具体一点要怎样改呢？