Python confusion_matrix_fnの例

コード例 #1

ファイル: metrics.py プロジェクト: 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

コード例 #2

ファイル: misc.py プロジェクト: 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