Bert-Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Bert-Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

输出结果各分类概率值问题

Open shengtaovvv opened this issue 3 years ago • 2 comments

请问模型预测文本输出的各分类概率值是如何计算的，我想要对结果的各个分类概率值进行0-1归一化处理，但是不知道这个该如何转化 image

shengtaovvv avatar May 12 '22 09:05 shengtaovvv

自己在这里最后加一个softmax层就好了

taosir avatar May 17 '22 04:05 taosir

自己在这里最后加一个softmax层就好了

非常感谢，已经解决了，就是这么操作的

shengtaovvv avatar May 18 '22 07:05 shengtaovvv