如何使用keras.utils.to_categorical检查一个热标签的功能名称?

时间:2019-05-14 05:51:09

标签: python keras

我想在应用keras.util.to_categorical后检查词汇。

例如。

from keras.utils import to_categorical
l = to_categorical(np.asarray([1,2,3,4,5,6]))

当我打印l [0]

我的结果:

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

但是我不知道这个结果代表什么?是1还是2还是3还是???

我想知道的是哪个索引代表我的numpy数组中的哪个值

请帮助我,谢谢。

1 个答案:

答案 0 :(得分:1)

编码是固定的位置一键编码。 0在第一栏中将始终为1,1在第二栏中将始终为1,依此类推。使用示例更好地理解-

from keras.utils import to_categorical
import numpy as np

arr1 = np.asarray([0,1,2])
arr2 = np.asarray([0,5,2])
arr3 = np.asarray([1,3,8])

print("arr1")
print(to_categorical(arr1))
print("arr2")
print(to_categorical(arr2))
print("arr3")
print(to_categorical(arr3))

输出:

arr1
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
arr2
[[1. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 1.]
[0. 0. 1. 0. 0. 0.]]
arr3
[[0. 1. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 1. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 1.]]