python - to_categorical 返回 2 列矩阵

Question

我使用 keras 中提供的 to_categorical 函数将浮点类型的 numpy ndarray 转换为其对应的二进制。 Y 的尺寸为 2144x1，但函数返回的数组尺寸为 2144x2。如何从 to_categorical 获取 2144x1 数组？Y 仅包含 0 和 1(浮点型)函数调用:

y_binary = to_categorical(Y)

Answer 1

keras.utils.to_categorical(y, num_classes=None)

将类向量(整数)转换为二进制类矩阵。

但它看起来并不支持 float (它对所有数字都做 floor)

to_categorical([0, 0.1, 0.2, 0.3, 1])
[[ 1.  0.]
 [ 1.  0.]
 [ 1.  0.]
 [ 1.  0.]
 [ 0.  1.]]


to_categorical([0, 1, 2, 3, 1])
[[ 1.  0.  0.  0.]
 [ 0.  1.  0.  0.]
 [ 0.  0.  1.  0.]
 [ 0.  0.  0.  1.]
 [ 0.  1.  0.  0.]]

一种解决方案是，将其放大为正整数并转换为分类。