在使用类似 keras.np_utils.to_categorical 的函数,对标签转换成one-hot编码的时候,要使标签从0开始。即如果是两类,设标签里面的内容为0和1,不能设置为1和2或其他数字。
比如:
from keras.utils import np_utils
x=[]
x.append(2)
x.append(1)
print(x)
x = np_utils.to_categorical(x, 2)
print(x)
这就是错误的
from keras.utils import np_utils
x=[]
x.append(0)
x.append(1)
print(x)
x = np_utils.to_categorical(x, 2)
print(x)
这样才正确