From 052cefb5c5eadf7dc991c9ebdf1b83f7255da5f7 Mon Sep 17 00:00:00 2001
From: Randall Britten <3136161+vertexcite@users.noreply.github.com>
Date: Fri, 3 Nov 2023 11:12:46 +1300
Subject: [PATCH] Fix classification_privacy.ipynb

Fixed by calling `compute_dp_sgd_privacy_statement`.
---
 g3doc/tutorials/classification_privacy.ipynb | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/g3doc/tutorials/classification_privacy.ipynb b/g3doc/tutorials/classification_privacy.ipynb
index 233656c6..ee14c61f 100644
--- a/g3doc/tutorials/classification_privacy.ipynb
+++ b/g3doc/tutorials/classification_privacy.ipynb
@@ -417,11 +417,12 @@
       },
       "outputs": [],
       "source": [
-        "compute_dp_sgd_privacy.compute_dp_sgd_privacy(n=train_data.shape[0],\n",
+        "print(compute_dp_sgd_privacy.compute_dp_sgd_privacy_statement(\n",
+        "                                              number_of_examples=train_data.shape[0],\n",
         "                                              batch_size=batch_size,\n",
         "                                              noise_multiplier=noise_multiplier,\n",
-        "                                              epochs=epochs,\n",
-        "                                              delta=1e-5)"
+        "                                              num_epochs=epochs,\n",
+        "                                              delta=1e-5))"
       ]
     },
     {
@@ -430,7 +431,7 @@
         "id": "c-KyttEWFRDc"
       },
       "source": [
-        "The tool reports that for the hyperparameters chosen above, the trained model has an $\\epsilon$ value of 1.18."
+        "The tool reports that for the hyperparameters chosen above, the trained model has an $\\epsilon$ value of 3.426."
       ]
     },
     {