Part1 of image classifier

import-data.py

# importing libraries
import numpy as np
from tensorflow import keras
from keras.datasets import mnist

#loading dataset
(train_X, train_y), (val_X, val_y) = mnist.load_data()

#normalizing the dataset
train_X, val_X = train_X/255, val_X/255

# visualizing 9 rndom digits from the dataset
for i in range(331,340):
    plt.subplot(i)
    a = np.random.randint(0, train_X.shape[0], 1)
    plt.imshow(train_X[a[0]], cmap = plt.get_cmap('binary'))

plt.tight_layout()
plt.show()