fujiyuu75
11/3/2019 - 10:56 AM

display one image with heatmap in matplotlib.

plt.figure()
plt.imshow(train_images[0])
plt.colorbar()
plt.grid(False)
plt.show()