schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add kwargs to constructor of custom model

main
Aurélien Geron 2019-02-28 19:48:50 +08:00
parent ed5c7d6db8
commit e938df8fb9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
10_neural_nets_with_keras.ipynb
View File

@ -1054,8 +1054,8 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"class WideAndDeepModel(keras.models.Model):\n", "class WideAndDeepModel(keras.models.Model):\n",
" def __init__(self, units=30, activation=\"relu\"):\n", " def __init__(self, units=30, activation=\"relu\", **kwargs):\n",
" super().__init__()\n", " super().__init__(**kwargs)\n",
" self.hidden1 = keras.layers.Dense(units, activation=activation)\n", " self.hidden1 = keras.layers.Dense(units, activation=activation)\n",
" self.hidden2 = keras.layers.Dense(units, activation=activation)\n", " self.hidden2 = keras.layers.Dense(units, activation=activation)\n",
" self.main_output = keras.layers.Dense(1)\n", " self.main_output = keras.layers.Dense(1)\n",