626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Updates to the membership inference codelab.

Browse source 
PiperOrigin-RevId: 318049409
This commit is contained in:
A. Unique TensorFlower 2020-06-24 05:56:02 -07:00
parent c07e87d0b5
commit e40a35f9c0
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tensorflow_privacy/privacy/membership_inference_attack/codelab.ipynb
View file

@ -101,7 +101,6 @@
" pass\n", " pass\n",
"\n", "\n",
"import numpy as np\n", "import numpy as np\n",
"from colabtools import adhoc_import\n",
"from typing import Tuple, Text\n", "from typing import Tuple, Text\n",
"from scipy import special\n", "from scipy import special\n",
"\n", "\n",
@ -132,7 +131,7 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"!pip install tensorflow_privacy\n", "!pip3 install git+https://github.com/tensorflow/privacy\n",
"\n", "\n",
"from tensorflow_privacy.privacy.membership_inference_attack import membership_inference_attack as mia" "from tensorflow_privacy.privacy.membership_inference_attack import membership_inference_attack as mia"
] ]
@ -252,8 +251,8 @@
"cce = tf.keras.backend.categorical_crossentropy\n", "cce = tf.keras.backend.categorical_crossentropy\n",
"constant = tf.keras.backend.constant\n", "constant = tf.keras.backend.constant\n",
"\n", "\n",
"loss_train = cce(constant(y_train), prob_train, from_logits=False).numpy()\n", "loss_train = cce(constant(y_train), constant(prob_train), from_logits=False).numpy()\n",
"loss_test = cce(constant(y_test), prob_test, from_logits=False).numpy()" "loss_test = cce(constant(y_test), constant(prob_test), from_logits=False).numpy()"
] ]
}, },
{ {