How to get predicted class labels in TensorFlow's MNIST example?

前端 未结 2 1821
心在旅途
心在旅途 2021-01-03 11:53

I am new to Neural Networks and went through the MNIST example for beginners.

I am currently trying to use this example on another dataset from Kaggle that does not

2条回答
  •  失恋的感觉
    2021-01-03 12:28

    prediction=tf.argmax(y,1)
    print prediction.eval(feed_dict={x: mnist.test.images}).
    

    For more details, check out https://github.com/tensorflow/tensorflow/issues/97

提交回复
热议问题