From 3a15ab71cba3e7b2ec006448af681cc9483e999e Mon Sep 17 00:00:00 2001 From: bobarna Date: Fri, 3 Dec 2021 12:24:21 +0100 Subject: [PATCH] Change deprecated 'lr' argument to 'learning_rate' This gets rid of the UserWarning: `The 'lr' argument is deprecated, use 'learning_rate' instead.` --- intro-teaser.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/intro-teaser.ipynb b/intro-teaser.ipynb index 8ea69b9..2c03e72 100644 --- a/intro-teaser.ipynb +++ b/intro-teaser.ipynb @@ -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", @@ -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)" ] },