mirror of
https://github.com/ArthurDanjou/handson-ml3.git
synced 2026-01-14 12:14:36 +01:00
Correct small "code typo"
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user