From 3658ef5dbce16f50dcf86a3e026967676b0f80bb Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 24 Jun 2020 07:44:21 -0700 Subject: [PATCH] Updates to the membership inference codelab. PiperOrigin-RevId: 318061869 --- .../membership_inference_attack/membership_inference_attack.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_privacy/privacy/membership_inference_attack/membership_inference_attack.py b/tensorflow_privacy/privacy/membership_inference_attack/membership_inference_attack.py index 42d5d42..7d130fd 100644 --- a/tensorflow_privacy/privacy/membership_inference_attack/membership_inference_attack.py +++ b/tensorflow_privacy/privacy/membership_inference_attack/membership_inference_attack.py @@ -633,7 +633,7 @@ def run_all_attacks_and_create_summary( """ summary = [] metrics = ['auc', 'advantage'] - attack_classifiers = ['lr', 'rf', 'mlp', 'knn'] + attack_classifiers = ['lr', 'knn'] results = run_all_attacks( loss_train, loss_test,