Skip to content

Commit d7ce21f

Browse files
authored
Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (tensorflow#5717)
This tag should match EVAL_HP_NUM_NEG.
1 parent 541d584 commit d7ce21f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

official/recommendation/data_preprocessing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def _filter_index_sort(raw_rating_path, match_mlperf):
152152
num_items = len(original_items)
153153

154154
mlperf_helper.ncf_print(key=mlperf_helper.TAGS.PREPROC_HP_NUM_EVAL,
155-
value=num_users * (1 + rconst.NUM_EVAL_NEGATIVES))
155+
value=rconst.NUM_EVAL_NEGATIVES)
156156
mlperf_helper.ncf_print(
157157
key=mlperf_helper.TAGS.PREPROC_HP_SAMPLE_EVAL_REPLACEMENT,
158158
value=match_mlperf)

0 commit comments

Comments
 (0)