import tensorflow as tf
import os
mnist = tf.keras.datasets.mnist
from matplotlib import pyplot as plt

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train/255.0, x_test/255.0
print(y_test)

plt.imshow(x_train[0], cmap="gray")
plt.show()