嘿,我正在试图弄清楚如何获得CIFAR示例中每个类的概率.目前我可以在输入张量中输入预测的类并返回真或假,但它不会告诉我它认为的类(满分为10)以及它对该事实的概率.有关如何在sess.run()之后获取此信息的任何想法？
用于评估(in cifar10_eval.py)的CIFAR-10示例代码产生(作为中间输出)logits张量,该张量包含给定图像在每个类中的对数概率.传递此张量sess.run()将为您提供必要的信息,以告诉您图像的预测类是什么.如果您计算tf.nn.softmax(logits)并传递给它sess.run(),您将获得每个类的概率(范围从0到1).