Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torch-cluster.git" did not exist on "c0e7bec93baf03b05fd0b30643868175f1c6f8dd"
Unverified Commit d7ce21fa authored by Reed's avatar Reed Committed by GitHub
Browse files

Fix PREPROC_HP_NUM_EVAL tag for MLPerf. (#5717) 

This tag should match EVAL_HP_NUM_NEG.
parent 541d584e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/data_preprocessing.py
View file @ d7ce21fa
...@@ -152,7 +152,7 @@ def _filter_index_sort(raw_rating_path, match_mlperf): ...@@ -152,7 +152,7 @@ def _filter_index_sort(raw_rating_path, match_mlperf):
num_items = len(original_items) num_items = len(original_items)
mlperf_helper.ncf_print(key=mlperf_helper.TAGS.PREPROC_HP_NUM_EVAL, 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( mlperf_helper.ncf_print(
key=mlperf_helper.TAGS.PREPROC_HP_SAMPLE_EVAL_REPLACEMENT, key=mlperf_helper.TAGS.PREPROC_HP_SAMPLE_EVAL_REPLACEMENT,
value=match_mlperf) value=match_mlperf)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment