preprocess_dataset_fashion_mnist.py

from tensorflow.keras.utils import to_categorical

# Normalize the images
train_images = train_images / 255.0
test_images = test_images / 255.0

# Reshape them for our CNN model
train_images = train_images.reshape(60000,28,28,1)
test_images = test_images.reshape(10000,28,28,1)
print(train_images.shape)

# Encode labels to one-hot
train_labels = to_categorical(train_labels)
test_labels = to_categorical(test_labels)
print(train_labels[0])