In [1]:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
from tensorflow.keras.datasets import fashion_mnist
import matplotlib.pyplot as plt
(X_train, y_train), (X_test, y_test) = fashion_mnist.load_data()

In [2]:
plt.rcParams['figure.figsize'] = (7,7)
offset = 0
labels = ["T-shirt/top", "Trouser", "Pullover", "Dress",
          "Coat", "Sandal", "Shirt", "Sneaker", "Bag", "Ankle boot"]
for i in range(9):
    plt.subplot(3,3,i+1)
    plt.imshow(X_train[i+offset], cmap='gray', interpolation='none')
    plt.title(labels[y_train[i+offset]])
    plt.axis('off')



In [ ]: