ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

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