我试图预测28000 MNIST图片的类别,但无论list(predict_results)
中的更改如何,input_fn
都保持不变。
以下是预测过程的代码:
mnist_classifier = tf.estimator.Estimator(model_fn = cnn_model_fn, model_dir = "/tmp/mnist_kaggle_model2")
for i in range(1,28000):
input_fn = tf.estimator.inputs.numpy_input_fn(
x = {'x': test_images[(i-1), ]},
num_epochs = 1,
shuffle = False
)
predict_results = mnist_classifier.predict(input_fn = input_fn)
print(list(predict_results))
以下是我改编test_images
的方式:
test_images = test_data.reshape(28000 , 28, 28, 1)
问题与我重塑test_images
的方式有关,还是与input_fn
一致?
=============================================== ============================
我完全对这个问题感到困惑,非常感谢任何帮助。
谢谢。