tensorflow.keras.utils.to_categorical - Python

to_categorical是由TensorFlow的Keras提供的一个工具，用于将类向量（整数）转换为二进制类矩阵。

用法

tensorflow.keras.utils.to_categorical(y, num_classes=None, dtype='float32')

参数

• y: 类向量或整数类型的数据，例如[1, 3, 2, 0]。如果 num_classes=None，则该函数通过数据推断分类数。
• num_classes: 总分类数。
• dtype: 输出数组的数据类型。

返回值

以num_classes大小的二进制矩阵形式返回输入的类向量。

示例

from tensorflow.keras.utils import to_categorical 

labels = [0, 1, 2, 3]
one_hot_labels = to_categorical(labels)

print(one_hot_labels)

# Output:
# array([[1., 0., 0., 0.],
#        [0., 1., 0., 0.],
#        [0., 0., 1., 0.],
#        [0., 0., 0., 1.]], dtype=float32)

在这个示例中，我们将一个类向量 [0, 1, 2, 3] 转换成了一个以大小为 4 的二进制矩阵的形式输出。函数自动推断了分类数为 4。