您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何在TensorFlow的MNIST示例中获取预测的类标签?

如何在TensorFlow的MNIST示例中获取预测的类标签?

我认为您只需要按照教程中所述评估输出张量即可:

accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))
print(sess.run(accuracy, Feed_dict={x: mnist.test.images, y_: mnist.test.labels}))

获取张量的输出,请参阅docs

在会话中启动图后,可以将Tensor的值传递给Session.run()来计算它的值。t.eval()是调用tf.get_default_session()。run(t)的快捷方式。

如果要获得预测而不是准确性,则需要以y相同的方式评估输出张量:

print(sess.run(y, Feed_dict={x: mnist.test.images}))
其他 2022/1/1 18:30:18 有366人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶