Source code for

.. module:: CMetricConfusionMatrix
   :synopsis: Confusion Matrix

.. moduleauthor:: Ambra Demontis <>

from sklearn.metrics import confusion_matrix
from secml.array import CArray
from import CMetric

[docs]class CMetricConfusionMatrix(CMetric): def _performance_score(self, y_true, y_pred): """Computes the Confusion Matrix. Parameters ---------- y_true : CArray Ground truth (true) labels or target scores. y_pred : CArray Predicted labels, as returned by a CClassifier. Returns ------- CArray Confusion matrix with shape = [n_classes, n_classes]. """ y_true = CArray(y_true) y_pred = CArray(y_pred) return CArray(confusion_matrix(y_true.tondarray(), y_pred.tondarray()))