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": {