Итак, я пытаюсь создать модель RNN для классификации в наборе данных MNIST. По умолчанию форма набора данных (60000, 28, 28). Я пытался использовать это в своем слое LSTM с
model.add (LSTM (единиц = 100, return_sequence = True, input_shape = (X_train.shape [1:]))) model.add (Dropout (0,2) ))
model.add (LSTM (единицы = 100, return_sequence = True)) model.add (выпадение (0,2))
model.add (LSTM (единицы = 100)) модель .add (выпадение (0,2))
model.add (плотное (единицы = 100)) model.add (выпадение (0,2))
model.add (плотное (единицы = 10) )
но выдает ошибку «Вход 0 несовместим со слоем lstm_13: ожидаемый ndim = 3 найден ndim = 2»
Я также пытался изменить форму с помощью: X_train = np.reshape (X_train, X_train.shape + (1,)) создание формы X_train.shape Но все равно не повезло.
Может ли кто-нибудь помочь мне с подробным объяснением того, как должен формироваться входной слой?
Большое спасибо