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

update data_structures_test

Browse source
This commit is contained in:
Liwei Song 2020-12-10 18:44:52 -05:00
parent d0d2108ad8
commit d6d70f6211
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -299,8 +299,8 @@ class SingleRiskScoreResultTest(absltest.TestCase):
train_risk_scores=np.array([0.91,1,0.92,0.82,0.75]),
test_risk_scores=np.array([0.81,0.7,0.75,0.25,0.3]))
self.assertEqual(result.attack_with_varied_thresholds([0.8,0.7])[1], [0.8,0.625])
self.assertEqual(result.attack_with_varied_thresholds([0.8,0.7])[2], [0.8,1])
self.assertEqual(result.attack_with_varied_thresholds(np.array([0.8,0.7]))[1].tolist(), [0.8,0.625])
self.assertEqual(result.attack_with_varied_thresholds(np.array([0.8,0.7]))[2].tolist(), [0.8,1])
class AttackResultsCollectionTest(absltest.TestCase):