def _compute_confusion_matrix(self): if self._version == "tnt": confusion_matrix = self.confusion_matrix.value() elif self._version == "sklearn": confusion_matrix = confusion_matrix_fn(y_true=self.targets, y_pred=self.outputs) return confusion_matrix
def _compute_confusion_matrix(self): if self._version == "tnt": confusion_matrix = self.confusion_matrix.value() elif self._version == "sklearn": confusion_matrix = confusion_matrix_fn(y_true=self.targets, y_pred=self.outputs) else: raise NotImplementedError() return confusion_matrix