text_plain_output_1.png
x_train = x_train.reshape(len(x_train), 200 * 200)
x_test = x_test.reshape(len(x_test), 200 * 200)
x_val = x_val.reshape(len(x_val), 200 * 200)
print(x_train.shape)
print(x_test.shape)
print(x_val.shape)