Подготовка переменной 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)