set tf seed

7b6c8999 · guptapriya · guptapriya · c80de2ca · 7b6c8999
Commit 7b6c8999 authored Jun 06, 2019 by guptapriya Committed by guptapriya Jun 12, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

official/recommendation/ncf_keras_main.py official/recommendation/ncf_keras_main.py +4 -0

No files found.
--- a/official/recommendation/ncf_keras_main.py
+++ b/official/recommendation/ncf_keras_main.py
@@ -245,6 +245,10 @@ def _get_keras_model(params):


 def run_ncf(_):
+  if FLAGS.seed is not None:
+    print("Setting tf seed")
+    tf.random.set_seed(FLAGS.seed)
+  
  """Run NCF training and eval with Keras."""
  # TODO(seemuch): Support different train and eval batch sizes
  if FLAGS.eval_batch_size != FLAGS.batch_size: