Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (tensorflow#5717)

This tag should match EVAL_HP_NUM_NEG.
AndrewRWilliams · Nov 7, 2018 · d7ce21f · d7ce21f
1 parent 541d584
commit d7ce21f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/official/recommendation/data_preprocessing.py b/official/recommendation/data_preprocessing.py
@@ -152,7 +152,7 @@ def _filter_index_sort(raw_rating_path, match_mlperf):
   num_items = len(original_items)
 
   mlperf_helper.ncf_print(key=mlperf_helper.TAGS.PREPROC_HP_NUM_EVAL,
-                          value=num_users * (1 + rconst.NUM_EVAL_NEGATIVES))
+                          value=rconst.NUM_EVAL_NEGATIVES)
   mlperf_helper.ncf_print(
       key=mlperf_helper.TAGS.PREPROC_HP_SAMPLE_EVAL_REPLACEMENT,
       value=match_mlperf)