keras.utils.to_categorical方法

to_categorical(y, num_classes=None, dtype=‘float32’)

将整型标签转为onehot。y为int数组，维度是(m, ), num_classes为标签类别总数n，默认是不指定（索引从0开始的）。

返回：如果num_classes=None，返回len(y)* [max(y)+1]（维度为(m,n)），否则为
len(y)* num_classes。
废话少说，放码过来：

import keras

a=keras.utils.to_categorical([1,2,3])
print('a=',a)
>>>>
a= [[0. 1. 0. 0.]
   [0. 0. 1. 0.]
   [0. 0. 0. 1.]]

b=keras.utils.to_categorical([1,2,3],num_classes=5)
print('b=',b)
>>>>
>b= [[0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]]