amirhfarzaneh
2/23/2018 - 9:02 AM

vgg13-mnist-sample.py

index = 7
plt.imshow(train_data[index].reshape(28, 28))
print ("y = " + str(np.squeeze(train_labels[index])))