【发布时间】:2016-06-24 02:38:10
【问题描述】:
嘿,我正试图弄清楚如何获取 CIFAR 示例中每个类的概率。目前我可以在输入张量中输入预测的类并返回一个真或假,但它不会告诉我它认为的类(满分 10)以及它对这一事实的概率。任何想法如何在 sess.run() 之后获取此信息?
【问题讨论】:
标签: tensorflow
嘿,我正试图弄清楚如何获取 CIFAR 示例中每个类的概率。目前我可以在输入张量中输入预测的类并返回一个真或假,但它不会告诉我它认为的类(满分 10)以及它对这一事实的概率。任何想法如何在 sess.run() 之后获取此信息?
【问题讨论】:
标签: tensorflow
用于评估的 CIFAR-10 示例代码(在 cifar10_eval.py 中)生成(作为中间输出)logits 张量,其中包含给定图像在每个类中的对数概率。将此张量传递给sess.run() 将为您提供必要的信息,告诉您图像的预测类别是什么。相反,如果您计算 tf.nn.softmax(logits) 并将其传递给 sess.run(),您将获得每个类的概率(范围从 0 到 1)。
【讨论】: