Merge pull request #2256 from vaharoni:vaharoni/text-classification-fix

PiperOrigin-RevId: 578942418
tensorflow · Nov 2, 2023 · 60abeb0 · 60abeb0
2 parents eb0f9bf + a0901ae
commit 60abeb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/site/en/tutorials/keras/text_classification.ipynb b/site/en/tutorials/keras/text_classification.ipynb
@@ -631,7 +631,7 @@
       "outputs": [],
       "source": [
         "model = tf.keras.Sequential([\n",
-        "  layers.Embedding(max_features + 1, embedding_dim),\n",
+        "  layers.Embedding(max_features, embedding_dim),\n",
         "  layers.Dropout(0.2),\n",
         "  layers.GlobalAveragePooling1D(),\n",
         "  layers.Dropout(0.2),\n",