0

我在训练 BiLSTM 模型时遇到以下错误并且找不到任何解决方案。

ValueError: Error when checking input: expected embedding_input to have 2 dimensions, but got array with shape (298, 200, 1)

以下是我的项目中构建的架构。请帮我。

    model.add(tf.keras.layers.Embedding(input_dim=n_vocab, output_dim=128, batch_input_shape=(maxlen, 1)))
    model.add(tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64)))
    model.add(tf.keras.layers.Dropout(0.5))
    model.add(tf.keras.layers.Dense(n_vocab, activation='sigmoid'))```
4

0 回答 0