tf.compat.v1.confusion_matrix

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

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.math.confusion_matrix

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

Столбцы матрицы соответствуют меткам предсказаний, а строки — фактическим меткам. Матрица ошибок всегда представляет собой двумерный массив формы [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], в результате чего получается матрица ошибок размером 5×5.

Аргументы
`labels`	Одномерный `Tensor` фактических меток для задачи классификации.
`predictions`	Одномерный `Tensor` предсказаний для данной задачи классификации.
`num_classes`	Возможное число меток в задаче классификации. Если это значение не указано, оно будет вычислено с использованием массивов предсказаний и меток.
`dtype`	Тип данных матрицы ошибок.
`name`	Имя области видимости.
`weights`	Необязательный `Tensor`, форма которого совпадает с `predictions`.

Возвращает
`Tensor` типа `dtype` формы `[n, n]`, представляющий матрицу ошибок, где `n` — число возможных меток в задаче классификации.

Вызывает исключения
`ValueError`	Если предсказания и метки не являются одномерными векторами и имеют несовпадающие формы либо если `weights` не равно `None` и его форма не совпадает с `predictions`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/confusion_matrix