Fix Spectral-normalized Neural Gaussian Process tutorial build

8bitmp3 · copybara-github · commit f1e668fc919e · 2022-12-21T13:18:48.000-08:00
PiperOrigin-RevId: 496991390
diff --git a/site/en/tutorials/understanding/sngp.ipynb b/site/en/tutorials/understanding/sngp.ipynb
@@ -143,7 +143,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install --use-deprecated=legacy-resolver tf-models-official"
+        "!pip install -U -q --use-deprecated=legacy-resolver tf-models-official tensorflow"
       ]
     },
     {
@@ -480,7 +480,7 @@
       "source": [
         "loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)\n",
         "metrics = tf.keras.metrics.SparseCategoricalAccuracy(),\n",
-        "optimizer = tf.keras.optimizers.Adam(learning_rate=1e-4)\n",
+        "optimizer = tf.keras.optimizers.legacy.Adam(learning_rate=1e-4)\n",
         "\n",
         "train_config = dict(loss=loss, metrics=metrics, optimizer=optimizer)"
       ]

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,7 @@`
`143`	`143`	`},`
`144`	`144`	`"outputs": [],`
`145`	`145`	`"source": [`
`146`		`- "!pip install --use-deprecated=legacy-resolver tf-models-official"`
	`146`	`+ "!pip install -U -q --use-deprecated=legacy-resolver tf-models-official tensorflow"`
`147`	`147`	`]`
`148`	`148`	`},`
`149`	`149`	`{`
`@@ -480,7 +480,7 @@`
`480`	`480`	`"source": [`
`481`	`481`	`"loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)\n",`
`482`	`482`	`"metrics = tf.keras.metrics.SparseCategoricalAccuracy(),\n",`
`483`		`- "optimizer = tf.keras.optimizers.Adam(learning_rate=1e-4)\n",`
	`483`	`+ "optimizer = tf.keras.optimizers.legacy.Adam(learning_rate=1e-4)\n",`
`484`	`484`	`"\n",`
`485`	`485`	`"train_config = dict(loss=loss, metrics=metrics, optimizer=optimizer)"`
`486`	`486`	`]`