How to solve that problem?
enter image description here
train_images = train_images / 255.0 model = keras.Sequential([ keras.layers.Flatten(input_