Apply suggestions from code review

Use `return_dict=True` for evaluate.
tensorflow · Aug 29, 2024 · bd582f3 · bd582f3
1 parent 70c03a9
commit bd582f3
Showing 1 changed file with 2 additions and 2 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)"
       ]
     },
     {