python - Keras np_utils.to_categorical 行为不同

Question

为什么 Keras to_categorical 在 [1, -1] 和 [2, -2] 上的行为不同？

y = [1, -1, -1]
y_ = np_utils.to_categorical(y)
array([[ 0.,  1.],
       [ 0.,  1.],
       [ 0.,  1.]])


y = [2, -2, -2]
y_ = np_utils.to_categorical(y)
array([[ 0.,  0.,  1.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.]])

Answer 1

to_categorical 不取负值，如果你的数据集有负值，你可以将 y - y.min() 传递给 to_categorical 所以它会像你期望的那样工作:

>>> y = numpy.array([2, -2, -2])
>>> to_categorical(y)
array([[ 0.,  0.,  1.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.]])
>>> to_categorical(y - y.min())
array([[ 0.,  0.,  0.,  0.,  1.],
       [ 1.,  0.,  0.,  0.,  0.],
       [ 1.,  0.,  0.,  0.,  0.]])