Блог

Я обучаю модель DNN на python. и данные выглядят следующим образом.

И я просто обучаю модель следующим образом;

 model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')])

model.compile(optimizer='adam',
              loss='sparse_cate&orical_crossentropy',
              metrics=['accuracy'])
model.fit(X_train, y_train, epochs=5)
  

Но ошибки происходят следующим образом

Ошибка при проверке входных данных: ожидалось, что flatten_input будет иметь 3 измерения, но получен массив с формой (940682, 2)

Как я могу решить эту проблему?

А вот форма X_train , y_train .