An Introduction to Implementing Neural Networks using TensorFlow

@klaptor wrote:

in the ** To test our model with our own eyes, let’s visualize its predictions** part i am always getting prediction is: 0

img_name = rng.choice(test.filename)
filepath = os.path.join(data_dir,'Train','Images','test',img_name)
img = imread(filepath, flatten = True)
test_index = int(img_name.split('.')[0]) - 49000
print ("Prediction is: {}".format(pred[test_index])) 
pylab.imshow(img,cmap = 'gray') 
pylab.show()

How do i solve it?

Posts: 1

Participants: 1

Read full topic