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.

PiperOrigin-RevId: 390412713
2021-08-12 11:02:46 -07:00 · 2021-08-12 11:02:46 -07:00 · 50673fec40
commit 50673fec40
parent b8c1ba72cd
1 changed files with 3 additions and 1 deletions
--- a/tensorflow_privacy/privacy/dp_query/tree_aggregation.py
+++ b/tensorflow_privacy/privacy/dp_query/tree_aggregation.py
@ -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: