626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

update privacy risk score code

Browse source
This commit is contained in:
Liwei Song 2020-12-16 16:01:29 -05:00
parent a4d108f270
commit 59bccb3a82
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/data_structures.py
View file

@ -471,7 +471,7 @@ class SingleRiskScoreResult:
precision_list = []
recall_list = []
meaningful_threshold_list = []
max_risk_score = max(train_risk_scores.max(), test_risk_scores.max())
max_risk_score = max(self.train_risk_scores.max(), self.test_risk_scores.max())
for threshold in threshold_list:
if threshold <= max_risk_score:
idx = np.argwhere(thresholds>=threshold)[-1][0]