Code changes to get ready for an incoming Keras optimizer migration.

Because this code is still testing TF1, we should use the legacy optimizer.

PiperOrigin-RevId: 473298980

Author: chenmoneygithub

site/en/guide/migrate/canned_estimators.ipynb

@@ -196,15 +196,15 @@
       "source": [
         "def create_sample_optimizer(tf_version):\n",
         "  if tf_version == 'tf1':\n",
-        "    optimizer = lambda: tf.keras.optimizers.Ftrl(\n",
+        "    optimizer = lambda: tf.keras.optimizers.legacy.Ftrl(\n",
         "        l1_regularization_strength=0.001,\n",
         "        learning_rate=tf1.train.exponential_decay(\n",
         "            learning_rate=0.1,\n",
         "            global_step=tf1.train.get_global_step(),\n",
         "            decay_steps=10000,\n",
         "            decay_rate=0.9))\n",
         "  elif tf_version == 'tf2':\n",
-        "    optimizer = tf.keras.optimizers.Ftrl(\n",
+        "    optimizer = tf.keras.optimizers.legacy.Ftrl(\n",
         "        l1_regularization_strength=0.001,\n",
         "        learning_rate=tf.keras.optimizers.schedules.ExponentialDecay(\n",
         "            initial_learning_rate=0.1, decay_steps=10000, decay_rate=0.9))\n",

site/en/tutorials/generative/style_transfer.ipynb

@@ -705,7 +705,7 @@
       },
       "outputs": [],
       "source": [
-        "opt = tf.keras.optimizers.Adam(learning_rate=0.02, beta_1=0.99, epsilon=1e-1)"
+        "opt = tf.keras.optimizers.legacy.Adam(learning_rate=0.02, beta_1=0.99, epsilon=1e-1)"
       ]
     },
     {