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

Add comment to explain support for LR

Browse source
This commit is contained in:
Aadyaa Maddi 2020-11-13 12:20:30 +08:00 committed by GitHub
parent bfc5ef333a
commit 641c4dd98c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -188,6 +188,10 @@ def run_seq2seq_attack(attack_input: Seq2SeqAttackInputData,
the attack result. the attack result.
""" """
attack_input.validate() attack_input.validate()
# The attacker uses the average rank (a single number) of a seq2seq dataset
# record to determine membership. Hence, only Logistic Regression is supported,
# as it makes the most sense for single-number features.
attacker = models.LogisticRegressionAttacker() attacker = models.LogisticRegressionAttacker()
prepared_attacker_data = models.create_seq2seq_attacker_data( prepared_attacker_data = models.create_seq2seq_attacker_data(