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,