Ошибка при запуске модели CNN-LSTM: ValueError: Вход 0 слоя lstm_13 несовместим со слоем: ожидаемый ndim = 3

Question

Я получаю сообщение об ошибке при выполнении кода ниже:

model = tf.keras.Sequential([
    tf.keras.layers.Embedding(vocab_size, embedding_dim, input_length=max_length),
    tf.keras.layers.Conv1D(128, 5, activation='relu'),
    tf.keras.layers.GlobalAveragePooling1D(),
    tf.keras.layers.LSTM(32),
    tf.keras.layers.Dense(6, activation='relu'),
    tf.keras.layers.Dense(2, activation='sigmoid')
   ])

Ошибка:

ValueError: Input 0 of layer lstm_13 is incompatible with the layer: expected ndim=3, found ndim=2. Full shape received: [None, 128]

Answer 1

изменение tf.keras.layers.GlobalAveragePooling1D() с tf.keras.layers.AveragePooling1D()