keras.to_categorical(y, num_classes=None, dtype=‘float32’)
作用:将标签转化为0nehot的形式
参数:
y:int型数组;
num_classes:标签类别个数

from keras.utils import to_categorical
label = [3]
l = to_categorical(label, num_classes=5)
print(l)

输出:[[0. 0. 0. 1. 0.]]
from keras.utils import to_categorical
label = [2,3]
l = to_categorical(label, num_classes=5)
print(l)

输出:[[0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]]
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐