Skip to content

Instantly share code, notes, and snippets.

@Samanehilchi
Samanehilchi / preparing_mnist.py
Created November 6, 2022 09:39 — forked from ogyalcin/preparing_mnist.py
Reshaping and Normalizing the MNIST Images
# Reshaping the array to 4-dims so that it can work with the Keras API
x_train = x_train.reshape(x_train.shape[0], 28, 28, 1)
x_test = x_test.reshape(x_test.shape[0], 28, 28, 1)
input_shape = (28, 28, 1)
# Making sure that the values are float so that we can get decimal points after division
x_train = x_train.astype('float32')
x_test = x_test.astype('float32')
# Normalizing the RGB codes by dividing it to the max RGB value.
x_train /= 255
x_test /= 255