Data preprocessing

def preprocessing(raw):
    label = tf.keras.utils.to_categorical(raw.label, 10)
    num_of_images = raw.shape[0]
    x_as_array = raw.values[:,1:]
    x_shaped_array = x_as_array.reshape(num_of_images, 28, 28, 1)
    image = x_shaped_array / 255
    return image, label

X, Y = preprocessing(train_fashion_data)
X_test, Y_test = preprocessing(test_fashion_data)