Correct small "code typo"
parent
a83d4885dc
commit
08e3870053
|
@ -2540,7 +2540,7 @@
|
||||||
"source": [
|
"source": [
|
||||||
"class BagOfWords(keras.layers.Layer):\n",
|
"class BagOfWords(keras.layers.Layer):\n",
|
||||||
" def __init__(self, n_tokens, dtype=tf.int32, **kwargs):\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",
|
" self.n_tokens = n_tokens\n",
|
||||||
" def call(self, inputs):\n",
|
" def call(self, inputs):\n",
|
||||||
" one_hot = tf.one_hot(inputs, self.n_tokens)\n",
|
" one_hot = tf.one_hot(inputs, self.n_tokens)\n",
|
||||||
|
|
Loading…
Reference in New Issue