python - 如何使用经过训练的 tensorflow 算法

Question

我已经安装了tensorflow并按照此处的教程进行操作

https://www.tensorflow.org/versions/0.6.0/tutorials/mnist/tf/index.html#tensorflow-mechanics-101

构建成功后，我可以得到相同大小数据集的评估结果，例如训练集为 1000X784，测试集为 1000X784。

但是如果我想使用上面训练的算法测试一个数据 1X784，并找出输出是什么，该怎么办？

我现在开始接触 tensorflow ，并且是机器学习的新手，我希望我已经描述了我自己。

Answer 1

我不清楚您在哪一部分遇到问题，但我认为您要问的是如何使用批量大小 1000 进行训练，但仅对单个输入进行预测。我假设您已经知道如何预测大小为 1000 的批处理。

如果模型输入占位符的第一个维度(通常是批量大小)设置为“无”，则在您提供输入时会推断该大小。因此，如果将 1000 更改为 None，则应该能够传递大小为 1 x 784 的输入来进行预测。