Skip to contents

Compute confusion matrix from predictions and true labels

Usage

calc_confusion_matrix(y_true, y_pred, labels = NULL)

Arguments

y_true

True class labels

y_pred

Predicted class labels

labels

Optional vector of labels to use (in order)

Value

A matrix with rows as true labels and columns as predicted labels

Examples

y_true <- c("A", "A", "B", "B", "C", "C")
y_pred <- c("A", "B", "B", "B", "C", "A")
calc_confusion_matrix(y_true, y_pred)
#>     Predicted
#> True A B C
#>    A 1 1 0
#>    B 0 2 0
#>    C 1 0 1