Searched refs:total_cm (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | metrics_impl.py | 255 total_cm = metric_variable(
276 update_op = state_ops.assign_add(total_cm, current_cm)
277 return total_cm, update_op
1135 total_cm, update_op = _streaming_confusion_matrix(labels, predictions,
1138 def compute_mean_iou(_, total_cm): argument
1141 math_ops.reduce_sum(total_cm, 0), dtypes.float32)
1143 math_ops.reduce_sum(total_cm, 1), dtypes.float32)
1144 cm_diag = math_ops.cast(array_ops.diag_part(total_cm), dtypes.float32)
1169 metrics_collections, compute_mean_iou, total_cm)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 2256 self.total_cm = self.add_weight(
2292 return self.total_cm.assign_add(current_cm)
2297 math_ops.reduce_sum(self.total_cm, axis=0), dtype=self._dtype)
2299 math_ops.reduce_sum(self.total_cm, axis=1), dtype=self._dtype)
2301 array_ops.diag_part(self.total_cm), dtype=self._dtype)
2319 K.set_value(self.total_cm, np.zeros((self.num_classes, self.num_classes)))
|
| D | metrics_test.py | 1955 self.assertArrayNear(self.evaluate(m_obj.total_cm)[0], [1, 0], 1e-1)
1956 self.assertArrayNear(self.evaluate(m_obj.total_cm)[1], [3, 0], 1e-1)
1958 self.assertArrayNear(self.evaluate(m_obj.total_cm)[0], [1, 0], 1e-1)
1959 self.assertArrayNear(self.evaluate(m_obj.total_cm)[1], [3, 0], 1e-1)
|