卷积神经网络预测一个类的概率

Question

对于这个example，我试图打印出＆＃34; tf.argmax（pred，1）＆＃34;，结果是[1 3 4 1 2]。

如何知道预测类别的概率？ Tensorflow是否提供任何API？

Answer 1

您可以做的是获得预测的argmax，并在此索引处获得预测。 这在numpy中更容易（例如，使用amax函数：https://docs.scipy.org/doc/numpy/reference/generated/numpy.amax.html）

tfpredicted,loss, acc = sess.run([pred,cost, accuracy], feed_dict={x: batch_x,y: batch_y,keep_prob: 1.})
print(np.amax(tfpredicted))

我没有测试代码，但希望它有效。否则你可能需要查看numpy提供的最大功能。

祝你好运！