daviddoji/pbdl-book
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #10 from bobarna/main

Browse Source 
Small updates to intro-teaser.ipynb
This commit is contained in:
Nils Thuerey 2021-12-03 12:54:51 +01:00 committed by GitHub
commit 24130d50be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
intro-teaser.ipynb
View File

@ -134,7 +134,7 @@
"# Neural network\n",
"act = tf.keras.layers.ReLU()\n",
"nn_sv = tf.keras.models.Sequential([\n",
" tf.keras.layers.Dense(10, activation=act),\n",
" tf.keras.layers.Dense(10, activation=act, input_shape=(1,)),\n",
" tf.keras.layers.Dense(10, activation=act),\n",
" tf.keras.layers.Dense(1,activation='linear')])"
]
@ -173,7 +173,7 @@
"source": [
"# Loss function\n",
"loss_sv = tf.keras.losses.MeanSquaredError()\n",
"optimizer_sv = tf.keras.optimizers.Adam(lr=0.001)\n",
"optimizer_sv = tf.keras.optimizers.Adam(learning_rate=0.001)\n",
"nn_sv.compile(optimizer=optimizer_sv, loss=loss_sv)\n",
"\n",
"# Training\n",
@ -264,7 +264,7 @@
"\n",
"# Model\n",
"nn_dp = tf.keras.models.Sequential([\n",
" tf.keras.layers.Dense(10, activation=act),\n",
" tf.keras.layers.Dense(10, activation=act, input_shape=(1,)),\n",
" tf.keras.layers.Dense(10, activation=act),\n",
" tf.keras.layers.Dense(1, activation='linear')])"
]
@ -291,7 +291,7 @@
"def loss_dp(y_true, y_pred):\n",
" return mse(y_true,y_pred**2)\n",
"\n",
"optimizer_dp = tf.keras.optimizers.Adam(lr=0.001)\n",
"optimizer_dp = tf.keras.optimizers.Adam(learning_rate=0.001)\n",
"nn_dp.compile(optimizer=optimizer_dp, loss=loss_dp)"
]
},