Skip to content

GitLab

"...ogb_lsc/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "caa6d6072c8344c4c44b9d0c9d7da7add75fcc96"
Unverified Commit c612d8c7 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

NCF benchmark: top_1 to hr_at_10_max. (#7291) 

* top_1 to hr_at_10_max.

* Call self._run_and_report_benchmark not Super
parent d09994b2
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ c612d8c7
...@@ -115,7 +115,7 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase): ...@@ -115,7 +115,7 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
Note: MLPerf like tests are not tuned to hit a specific hr@10 value, but Note: MLPerf like tests are not tuned to hit a specific hr@10 value, but
we want it recorded. we want it recorded.
""" """
super(NCFKerasAccuracy, self)._run_and_report_benchmark(hr_at_10_min=0.61) self._run_and_report_benchmark(hr_at_10_min=0.61)
def _run_and_report_benchmark(self, hr_at_10_min=0.630, hr_at_10_max=0.640): def _run_and_report_benchmark(self, hr_at_10_min=0.630, hr_at_10_max=0.640):
"""Run test and report results. """Run test and report results.
...@@ -128,8 +128,8 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase): ...@@ -128,8 +128,8 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
hr_at_10_max: Maximum acceptable hr@10 value. hr_at_10_max: Maximum acceptable hr@10 value.
""" """
super(NCFKerasAccuracy, self)._run_and_report_benchmark( super(NCFKerasAccuracy, self)._run_and_report_benchmark(
top_1_min=hr_at_10_min, hr_at_10_min=hr_at_10_min,
top_1_max=hr_at_10_max) hr_at_10_max=hr_at_10_max)
def benchmark_1_gpu_early_stop(self): def benchmark_1_gpu_early_stop(self):
self._setup() self._setup()
......
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