Add one to epochs in epsilon calculation

GitOrigin-RevId: 65eac5d94ce0366f560abd0614435312649f6e27
gretelai · Mar 8, 2022 · 90ae80a · 90ae80a
1 parent 34e62bd
commit 90ae80a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gretel_synthetics/tensorflow/train.py b/src/gretel_synthetics/tensorflow/train.py
@@ -71,7 +71,9 @@ def on_epoch_end(self, epoch, logs: dict = None):
         if self.config.dp:
             # Account for tf-privacy library writing to stdout
             with redirect_stdout(io.StringIO()):
-                eps, _ = compute_epsilon(self.num_examples_train, self.config, epoch)
+                eps, _ = compute_epsilon(
+                    self.num_examples_train, self.config, epoch + 1
+                )
                 logs[METRIC_EPSILON] = eps
 
             # NOTE: this is just a list of the same value, but