使用tf.nn.softmax而不是tf.argmax预测概率的问题

时间:2018-11-24 15:43:31

标签: python tensorflow

我已经搜索了这个问题,却找不到任何对我有帮助的答案。

我已经训练了一个模型,并使用tf.argmax(logits,-1)来预测正确的课程(在98个课程中)。效果很好,我成功预测了结果。

然后,我尝试查看导致该预测的概率,将tf.argmax命令替换为tf.nn.softmax。当我这样做时,它会预测另一个类的概率为0.9,而应该预测的类(即使用argmax时)的概率为0.3。这是错误的,因为用argmax预测的结果是正确的。为什么当我使用softmax时,概率选择另一个(不正确的)类?关于那里的问题有什么想法吗?预先谢谢你!

0 个答案:

没有答案