浅谈keras中的keras.utils.to_categorical用法
时间: 2023-05-21 22:07:22 浏览: 397
keras.utils.to_categorical和one hot格式解析
keras.utils.to_categorical函数用于将类向量(整数)转换为二进制类矩阵。该函数使用一个整数输入来生成一个独热编码输出。这通常用于将多类别标签转换为一个one-hot编码,以便在分类问题中使用。
举个例子,如果我们有一个大小为3的向量,其中元素的取值范围为0-2:
[0, 1, 2]
我们可以使用to_categorical函数将其转换为二进制类矩阵:
[[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]]
这个矩阵中的每一行都对应着一个元素,它们的索引值等于那个元素的数值。
注意:to_categorical函数返回一个NumPy数组。
阅读全文