Python confusion_matrix_fn Beispiele

Beispiel #1

Datei: metrics.py Projekt: murari-goswami/catalyst

 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

Beispiel #2

Datei: misc.py Projekt: vikasmech/catalyst

 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