当输入被馈送到估计器时,预测结果保持相同

时间:2018-06-06 06:38:00

标签: python tensorflow tensorflow-estimator

我试图预测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一致?

=============================================== ============================

我完全对这个问题感到困惑,非常感谢任何帮助。

谢谢。

0 个答案:

没有答案