tf.keras.utils.to_categorical
| View source on GitHub |
Преобразует вектор классов (целые числа) в бинарную матрицу классов.
tf.keras.utils.to_categorical(
y, num_classes=None, dtype='float32'
)
Например, для использования с categorical_crossentropy.
| Аргументы | |
|---|---|
y | Вектор классов, который нужно преобразовать в матрицу (целые числа от 0 до num_classes). |
num_classes | Общее количество классов. |
dtype | Тип данных, ожидаемый на входе. По умолчанию: 'float32'. |
| Возвращаемое значение | |
|---|---|
| Бинарное матричное представление входных данных. Ось классов расположена последней. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/utils/to_categorical
