Correct small "code typo"

2020-10-06 19:20:18 -04:00 · 2020-10-06 19:20:18 -04:00 · 08e3870053
parent a83d4885dc
commit 08e3870053
1 changed files with 1 additions and 1 deletions
--- a/13_loading_and_preprocessing_data.ipynb
+++ b/13_loading_and_preprocessing_data.ipynb
@ -2540,7 +2540,7 @@
   "source": [
    "class BagOfWords(keras.layers.Layer):\n",
    "    def __init__(self, n_tokens, dtype=tf.int32, **kwargs):\n",
-    "        super().__init__(dtype=tf.int32, **kwargs)\n",
+    "        super().__init__(dtype=dtype, **kwargs)\n",
    "        self.n_tokens = n_tokens\n",
    "    def call(self, inputs):\n",
    "        one_hot = tf.one_hot(inputs, self.n_tokens)\n",