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

Change deprecated 'lr' argument to 'learning_rate'

Browse Source 
This gets rid of the UserWarning:
`The 'lr' argument is deprecated, use 'learning_rate' instead.`
This commit is contained in:
bobarna
2021-12-03 12:24:21 +01:00
parent 6b347889f5
commit 3a15ab71cb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
intro-teaser.ipynb
View File

@@ -173,7 +173,7 @@
"source": [ "source": [
"# Loss function\n", "# Loss function\n",
"loss_sv = tf.keras.losses.MeanSquaredError()\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", "nn_sv.compile(optimizer=optimizer_sv, loss=loss_sv)\n",
"\n", "\n",
"# Training\n", "# Training\n",
@@ -291,7 +291,7 @@
"def loss_dp(y_true, y_pred):\n", "def loss_dp(y_true, y_pred):\n",
" return mse(y_true,y_pred**2)\n", " return mse(y_true,y_pred**2)\n",
"\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)" "nn_dp.compile(optimizer=optimizer_dp, loss=loss_dp)"
] ]
}, },