From d7ce21fa4d3b8b204530873ade75637e1313b760 Mon Sep 17 00:00:00 2001 From: Reed Date: Wed, 7 Nov 2018 10:31:47 -0800 Subject: [PATCH] Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (#5717) This tag should match EVAL_HP_NUM_NEG. --- official/recommendation/data_preprocessing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/recommendation/data_preprocessing.py b/official/recommendation/data_preprocessing.py index 329cde41888..24cac41b67e 100644 --- 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)