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

我是神经网络的新手,已经学习了适合初学者的MNIST示例。

我目前正在尝试将这个示例应用于来自Kaggle的另一个没有测试标签的数据集。

如果我在没有相应标签的测试数据集上运行模型,因此无法像MNIST示例中那样计算准确率,我希望能够查看预测结果。有没有办法访问观察值及其预测标签,并以一种美观的方式打印出来?


回答:

我认为你只需要像教程中所说的那样评估你的输出张量:

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

要获取张量的输出,请查看文档

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

如果你想要获取预测结果而不是准确率,你需要以相同的方式评估你的输出张量y

print(sess.run(y, feed_dict={x: mnist.test.images}))

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注