text_plain_output_1.png
img_rows, img_cols, channels = (28, 28, 1)
num_classes = 10
X_train = X_train / 255
X_test = X_test / 255

X_train = X_train.reshape((-1, img_rows, img_cols, channels))
X_test = X_test.reshape((-1, img_rows, img_cols, channels))
print('X_train set shape of {}'.format(X_train.shape))
print('X_test set shape of {}'.format(X_test.shape))