aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/metrics/python
diff options
context:
space:
mode:
authorGravatar Yan Facai (颜发才) <facai.yan@gmail.com>2018-09-11 21:35:22 +0800
committerGravatar Yan Facai (颜发才) <facai.yan@gmail.com>2018-09-12 14:25:26 +0800
commit2dd5fb6cfb16ccc612b6e278d6282ef90581c0bb (patch)
treefc370d5a5a0b43f3137b0a33a1c92dbb80753c1e /tensorflow/contrib/metrics/python
parentb2896c3cc3a0656b838f58975338d7dd309e3e62 (diff)
CLN: fix merge error
Diffstat (limited to 'tensorflow/contrib/metrics/python')
-rw-r--r--tensorflow/contrib/metrics/python/ops/metric_ops.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/metrics/python/ops/metric_ops.py b/tensorflow/contrib/metrics/python/ops/metric_ops.py
index 1ddd7e521b..d7c73c8f99 100644
--- a/tensorflow/contrib/metrics/python/ops/metric_ops.py
+++ b/tensorflow/contrib/metrics/python/ops/metric_ops.py
@@ -3904,8 +3904,8 @@ def cohen_kappa(labels,
total = math_ops.reduce_sum(pe_row)
pe_sum = math_ops.reduce_sum(
math_ops.div_no_nan(
- pe_row * pe_col,
- math_ops.maximum(total, 0),
+ math_ops.to_double(pe_row * pe_col),
+ math_ops.to_double(total),
name=None))
po_sum, pe_sum, total = (math_ops.to_double(po_sum),
math_ops.to_double(pe_sum),