我已经搜索了这个问题,却找不到任何对我有帮助的答案。
我已经训练了一个模型,并使用tf.argmax(logits,-1)来预测正确的课程(在98个课程中)。效果很好,我成功预测了结果。
然后,我尝试查看导致该预测的概率,将tf.argmax命令替换为tf.nn.softmax。当我这样做时,它会预测另一个类的概率为0.9,而应该预测的类(即使用argmax时)的概率为0.3。这是错误的,因为用argmax预测的结果是正确的。为什么当我使用softmax时,概率选择另一个(不正确的)类?关于那里的问题有什么想法吗?预先谢谢你!