为神经网络训练准备y_target变量

Question

我有一个y_target变量数组，其形状为(5000,)，我想将其用于Keras NN模型进行时间序列分类。

使用numpy newaxis已通过执行以下操作将尺寸从1D转换为2D：y_target = y_target[..., newaxis]

我的目标变量有2个类：[0,1]，所以最后一次转换的数组（形状为(5000,1)）。看起来像这样：

[[0],
 [0],
 [1],
 [0],
 [1]]

我想使用以下逻辑来变换该数组：(5000,2)将具有这样的二进制结构：

 [[1, 0],
  [1, 0],
  [0, 1],
  [1, 0],
  [0, 1]]

在大熊猫中有直接的方法吗？

Answer 1

我已经知道该怎么做了：）

一种可能的方法是使用Keras库和to_categorical()函数

from keras.utils import to_categorical
y_target = to_categorical(y_target)