keras.utils.to_categorical()
作用:将标签转化为0nehot的形式。num_classes:标签类别个数。
·
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.]]
更多推荐
所有评论(0)