ConfusionMatrix

class verta.data_types.ConfusionMatrix(value, labels)

Representation of a confusion matrix.

Parameters:

• value (list of list of float) – Cell values.

• labels (list of str) – Row/column labels.

Examples

from verta.data_types import ConfusionMatrix
data = ConfusionMatrix(
    value=[
        [6, 1],
        [2, 3],
    ],
    labels=["spam", "not spam"],
)
run.log_attribute("spam_confusion_matrix", data)