Skip to content

GitLab

Commit 042c9aaa authored by Toby Boyd's avatar Toby Boyd Committed by Shining Sun
Browse files

Ncf metric tweaks (#6633) 

* Use tf.image.resize_with_crop_or_pad

* exp_per_second and hr_at_10
parent 5c37e69c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ 042c9aaa
......@@ -105,8 +105,8 @@ class KerasNCFRealData(KerasNCFBenchmarkBase):
def _extract_benchmark_report_extras(self, stats):
extras = {}
extras['train_loss'] = stats['loss']
extras['eval_hit_rate'] = stats['eval_hit_rate']
extras['examples_per_second'] = stats['avg_exp_per_second']
extras['hr_at_10'] = stats['eval_hit_rate']
extras['exp_per_second'] = stats['avg_exp_per_second']
return extras
def benchmark_1_gpu(self):
......@@ -148,7 +148,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase):
def _extract_benchmark_report_extras(self, stats):
extras = {}
extras['examples_per_second'] = stats['avg_exp_per_second']
extras['exp_per_second'] = stats['avg_exp_per_second']
return extras
def benchmark_1_gpu(self):
......
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