tf.keras.utils.to_categorical

Преобразует вектор классов (целые числа) в бинарную матрицу классов.

tf.keras.utils.to_categorical(
    x, num_classes=None
)

Используется в блокнотах

Используется в руководствах
Реализация дифференциальной конфиденциальности с помощью TensorFlow Privacy Оценка рисков конфиденциальности с помощью отчёта о конфиденциальности TensorFlow Privacy Обучение на устройстве с TensorFlow Lite Классификация поз человека с помощью MoveNet и TensorFlow Lite Классификация CIFAR-10 с помощью XLA

Используется в руководствах

Например, для использования с categorical_crossentropy.

Аргументы
`x`	Массивоподобный объект со значениями классов, которые необходимо преобразовать в матрицу (целые числа от 0 до `num_classes - 1`).
`num_classes`	Общее число классов. Если `None`, это будет выведено как `max(x) + 1`. По умолчанию `None`.

Возвращаемое значение
Бинарное матричное представление входных данных в виде массива NumPy. Ось класса расположена последней.

Пример:

a = keras.utils.to_categorical([0, 1, 2, 3], num_classes=4)
print(a)
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

b = np.array([.9, .04, .03, .03,
              .3, .45, .15, .13,
              .04, .01, .94, .05,
              .12, .21, .5, .17],
              shape=[4, 4])
loss = keras.ops.categorical_crossentropy(a, b)
print(np.around(loss, 5))
[0.10536 0.82807 0.1011  1.77196]

loss = keras.ops.categorical_crossentropy(a, a)
print(np.around(loss, 5))
[0. 0. 0. 0.]

© 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/keras/utils/to_categorical