Change Embedding's input_dim argument

Wrong argument for the decoder's embedding layer.
tranqv · Oct 20, 2020 · 74afde5 · 74afde5
1 parent 215a7e0
commit 74afde5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/16_nlp_with_rnns_and_attention.ipynb b/16_nlp_with_rnns_and_attention.ipynb
@@ -3286,7 +3286,7 @@
  " len(INPUT_CHARS) + 1, encoder_embedding_size)(encoder_inputs)\n",
  "\n",
  "decoder_embedding_layer = keras.layers.Embedding(\n",
- " len(INPUT_CHARS) + 2, decoder_embedding_size)\n",
+ " len(OUTPUT_CHARS) + 2, decoder_embedding_size)\n",
  "decoder_embeddings = decoder_embedding_layer(decoder_inputs)\n",
  "\n",
  "encoder = keras.layers.LSTM(units, return_state=True)\n",
@@ -3582,7 +3582,7 @@
  " len(INPUT_CHARS) + 1, encoder_embedding_size)(encoder_inputs)\n",
  "\n",
  "decoder_embedding_layer = keras.layers.Embedding(\n",
- " len(INPUT_CHARS) + 2, decoder_embedding_size)\n",
+ " len(OUTPUT_CHARS) + 2, decoder_embedding_size)\n",
  "decoder_embeddings = decoder_embedding_layer(decoder_inputs)\n",
  "\n",
  "encoder = keras.layers.LSTM(units, return_state=True)\n",