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

Update remaining DPQuery tests to TF2.

Browse source 
PiperOrigin-RevId: 468793518
This commit is contained in:
Steve Chien 2022-08-19 15:07:53 -07:00 committed by A. Unique TensorFlower
parent fd64be5b5b
commit 9e25eee68b
1 changed files with 8 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tensorflow_privacy/privacy/dp_query/normalized_query_test.py
View file

@ -21,7 +21,6 @@ from tensorflow_privacy.privacy.dp_query import test_utils
class NormalizedQueryTest(tf.test.TestCase): class NormalizedQueryTest(tf.test.TestCase):
def test_normalization(self): def test_normalization(self):
with self.cached_session() as sess:
record1 = tf.constant([-6.0, 8.0]) # Clipped to [-3.0, 4.0]. record1 = tf.constant([-6.0, 8.0]) # Clipped to [-3.0, 4.0].
record2 = tf.constant([4.0, -3.0]) # Not clipped. record2 = tf.constant([4.0, -3.0]) # Not clipped.
@ -30,9 +29,8 @@ class NormalizedQueryTest(tf.test.TestCase):
numerator_query=sum_query, denominator=2.0) numerator_query=sum_query, denominator=2.0)
query_result, _ = test_utils.run_query(query, [record1, record2]) query_result, _ = test_utils.run_query(query, [record1, record2])
result = sess.run(query_result)
expected = [0.5, 0.5] expected = [0.5, 0.5]
self.assertAllClose(result, expected) self.assertAllClose(query_result, expected)
if __name__ == '__main__': if __name__ == '__main__':