We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents e3b7e01 + 5668c36 commit e976927Copy full SHA for e976927
1 file changed
TensorFlow2/Recommendation/WideAndDeep/trainer/utils/evaluator.py
@@ -265,7 +265,7 @@ def eval(self, step):
265
eval_data["streaming_map_val"] = np.around(map_value, 4)
266
267
for metric_name, metric in zip(self.metric_names, hand_reduced_metrics):
268
- eval_data[metric_name] = np.around(metric.result().numpy(), 4)
+ eval_data[metric_name] = np.around(metric.result().numpy().astype(np.float64), 4)
269
270
self.log(eval_data, current_step)
271
0 commit comments