ConfusionMatrix

Computes the confusion matrix from predictions and labels.

Abstract Signature:

ConfusionMatrix(labels: Tensor, predictions: Tensor, num_classes: int)

PyTorchJAX (Core)NumPyTensorFlow

PyTorch

API: torch.func.confusion_matrix

Strategy: Plugin (torch_confusion_matrix_impl)

JAX (Core)

API: —

Strategy: Plugin (jax_confusion_matrix_impl)

NumPy

API: sklearn.metrics.confusion_matrix

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.math.confusion_matrix

Strategy: Direct Mapping

Official Docs ↗