fix bugs (#969)

PaddlePaddle · Apr 20, 2021 · 7c66cbd · 7c66cbd
1 parent 455ef37
commit 7c66cbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb b/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb
@@ -171,10 +171,10 @@
     "def vectorizer(input, label=None, length=2000):\r\n",
     "    if label is not None:\r\n",
     "        for x, y in zip(input, label):\r\n",
-    "            yield np.array((x + [0]*length)[:2000]).astype('int64'), np.array([y]).astype('int64')\r\n",
+    "            yield np.array((x + [0]*length)[:length]).astype('int64'), np.array([y]).astype('int64')\r\n",
     "    else:\r\n",
     "        for x in input:\r\n",
-    "            yield np.array((x + [0]*length)[:2000]).astype('int64')"
+    "            yield np.array((x + [0]*length)[:length]).astype('int64')"
    ]
   },
   {