metrics.py 文件源码

def _compute_specific(self, predicted, targets):
        targets = tf.cast(targets, tf.int64)
        return tf.confusion_matrix(labels=targets,
                                   predictions=predicted,
                                   num_classes=self._num_classes)