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

Minor fix for the nondeterministic seed of tree_aggregation.GaussianNoiseGenerator. The previous log results won't change much, while one seed is probably good enough.

Browse source 
PiperOrigin-RevId: 390412713
This commit is contained in:
Zheng Xu 2021-08-12 11:02:46 -07:00 committed by A. Unique TensorFlower
parent b8c1ba72cd
commit 50673fec40
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -94,10 +94,12 @@ class GaussianNoiseGenerator(ValueGenerator):
An initial state.
"""
if self.seed is None:
time_now = tf.timestamp()
residual = time_now - tf.math.floor(time_now)
return tf.cast(
tf.stack([
tf.math.floor(tf.timestamp() * 1e6),
tf.math.floor(tf.math.log(tf.timestamp() * 1e6))
tf.math.floor(residual * 1e9)
]),
dtype=tf.int64)
else: