to_categorical就是将类别标签向量转换为二进制(只有0和1)的矩阵类型表示。每一个标签用矩阵的对应的行向量来表示。
np_utils.to_categorical(数据的类别标签, 总类别数)
例如类别的标签为0、1、2、3、4,将数据分为这5类,则经过to_categorical函数作用后。
0—1 0 0 0 0
1—0 1 0 0 0
2—0 0 1 0 0
3—0 0 0 1 0
4—0 0 0 0 1
to_categorical就是将类别标签向量转换为二进制(只有0和1)的矩阵类型表示。每一个标签用矩阵的对应的行向量来表示。
np_utils.to_categorical(数据的类别标签, 总类别数)
例如类别的标签为0、1、2、3、4,将数据分为这5类,则经过to_categorical函数作用后。
0—1 0 0 0 0
1—0 1 0 0 0
2—0 0 1 0 0
3—0 0 0 1 0
4—0 0 0 0 1