tf.math.confusion_matrix

Вычисляет матрицу ошибок классификации по предсказаниям и меткам.

tf.math.confusion_matrix(
    labels,
    predictions,
    num_classes=None,
    weights=None,
    dtype=tf.dtypes.int32,
    name=None
)

Столбцы матрицы представляют предсказанные метки, а строки — реальные метки. Матрица ошибок всегда является 2-мерным массивом формы [n, n], где n — количество допустимых меток для данной задачи классификации. И предсказания, и метки должны быть одномерными массивами одинаковой формы для работы этой функции.

Если num_classes равно None, то num_classes будет установлено в значение, равное единице плюс максимальное значение в массивах предсказаний или меток. Метки классов должны начинаться с 0. Например, если num_classes равно 3, то возможные метки будут [0, 1, 2].

Если weights не равно None, то каждое предсказание вносит свой соответствующий вес в итоговое значение ячейки матрицы ошибок.

Например:

tf.math.confusion_matrix([1, 2, 4], [2, 2, 4]) ==>
    [[0 0 0 0 0]
     [0 0 1 0 0]
     [0 0 1 0 0]
     [0 0 0 0 0]
     [0 0 0 0 1]]

Обратите внимание, что предполагается, что возможные метки равны [0, 1, 2, 3, 4], что приводит к матрице ошибок 5x5.