Результаты прогноза остаются теми же, когда ввод поступает в оценщик - PullRequest
Новая
       7

Результаты прогноза остаются теми же, когда ввод поступает в оценщик

0 голосов
/ 06 июня 2018

Я пытался предсказать классы 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?

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

Я совершенно запутался в проблеме, и любая помощь будет высоко оценена.

Спасибо.

...