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: 318061869
This commit is contained in:
A. Unique TensorFlower 2020-06-24 07:44:21 -07:00
parent 18fe9eb3aa
commit 3658ef5dbc
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow_privacy/privacy/membership_inference_attack/membership_inference_attack.py
View file

@ -633,7 +633,7 @@ def run_all_attacks_and_create_summary(
""" """
summary = [] summary = []
metrics = ['auc', 'advantage'] metrics = ['auc', 'advantage']
attack_classifiers = ['lr', 'rf', 'mlp', 'knn'] attack_classifiers = ['lr', 'knn']
results = run_all_attacks( results = run_all_attacks(
loss_train, loss_train,
loss_test, loss_test,