arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-29 19:20:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add kwargs to constructor of custom model

Browse Source
This commit is contained in:
Aurélien Geron
2019-02-28 19:48:50 +08:00
parent ed5c7d6db8
commit e938df8fb9
1 changed files with 2 additions and 2 deletions
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",