Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (#5717)

This tag should match EVAL_HP_NUM_NEG.

Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (#5717)
This tag should match EVAL_HP_NUM_NEG.
d7ce21fa · Reed · GitHub · 541d584e · d7ce21fa
Unverified Commit d7ce21fa authored Nov 07, 2018 by Reed Committed by GitHub Nov 07, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/recommendation/data_preprocessing.py official/recommendation/data_preprocessing.py +1 -1

No files found.
--- 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)