Merge pull request #2322 from sanskarmodi8:fix-tf-issue-74564

PiperOrigin-RevId: 669144821
tensorflow · Aug 30, 2024 · 3d45463 · 3d45463
2 parents 5c5a466 + bd582f3
commit 3d45463
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/site/en/tutorials/keras/text_classification.ipynb b/site/en/tutorials/keras/text_classification.ipynb
@@ -861,8 +861,8 @@
         ")\n",
         "\n",
         "# Test it with `raw_test_ds`, which yields raw strings\n",
-        "loss, accuracy = export_model.evaluate(raw_test_ds)\n",
-        "print(accuracy)"
+        "metrics = export_model.evaluate(raw_test_ds, return_dict=True)\n",
+        "print(metrics)"
       ]
     },
     {
@@ -969,7 +969,6 @@
     "accelerator": "GPU",
     "colab": {
       "name": "text_classification.ipynb",
-      "provenance": [],
       "toc_visible": true
     },
     "kernelspec": {