Fix flag name error in NCF create_ncf_data binary.

PiperOrigin-RevId: 264468498

Fix flag name error in NCF create_ncf_data binary.
PiperOrigin-RevId: 264468498
ec6ac5fa · A. Unique TensorFlower · a6f10ec0 · ec6ac5fa
Commit ec6ac5fa authored Aug 20, 2019 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/recommendation/create_ncf_data.py official/recommendation/create_ncf_data.py +2 -2

No files found.
--- a/official/recommendation/create_ncf_data.py
+++ b/official/recommendation/create_ncf_data.py
@@ -65,8 +65,8 @@ def prepare_raw_data(flag_obj):

  data_processing_params = {
      "train_epochs": flag_obj.num_train_epochs,
-      "batch_size": flag_obj.prebatch_size,
-      "eval_batch_size": flag_obj.prebatch_size,
+      "batch_size": flag_obj.train_prebatch_size,
+      "eval_batch_size": flag_obj.eval_prebatch_size,
      "batches_per_step": 1,
      "stream_files": True,
      "num_neg": flag_obj.num_negative_samples,