text_plain_output_1.png
import numpy as np

seed = 42
np.random.seed = seed

x_train = np.asarray(x_train, np.float32) / 255
x_test = np.asarray(x_test, np.float32) / 255
y_train = np.asarray(y_train)
y_test = np.asarray(y_test)

shape = 50 * 50 * 3
x_train = x_train.reshape(x_train.shape[0], shape)
x_test = x_test.reshape(x_test.shape[0], shape)

print('shape of new train data is :', x_train.shape)
print('shape of new Test data is :', x_test.shape)