From 08e387005399bba46e5b1aa605e467f47ef50272 Mon Sep 17 00:00:00 2001 From: Ian Beauregard Date: Tue, 6 Oct 2020 19:20:18 -0400 Subject: [PATCH] Correct small "code typo" --- 13_loading_and_preprocessing_data.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/13_loading_and_preprocessing_data.ipynb b/13_loading_and_preprocessing_data.ipynb index c258b82..7d7ff12 100644 --- 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",