From bd582f3330cdec34adab3867e4005014a90bbade Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Thu, 29 Aug 2024 13:53:21 -0700 Subject: [PATCH] Apply suggestions from code review Use `return_dict=True` for evaluate. --- site/en/tutorials/keras/text_classification.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/site/en/tutorials/keras/text_classification.ipynb b/site/en/tutorials/keras/text_classification.ipynb index 7b44c8e521..02e768d741 100644 --- 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)" ] }, {