Одна горячая кодировка и создание y_train для LSTM - PullRequest
Новая
       63

Одна горячая кодировка и создание y_train для LSTM

0 голосов
/ 17 марта 2020

У меня есть массив numpy, который содержит 54837488 значений

len(signal) = 54837488

Этот сигнал содержит 6877 деталей, которые помечены 9 различными метками (9 категориями). Метки классифицируются с помощью MultiLabelBinarizer (), поэтому каждая часть сигнала может быть отнесена к одной или нескольким из этих 9 меток.

В любом случае, вопрос заключается в том; как я могу создать y_train (горячее кодирование) для моего X_train (длина сигнала 54837488) и передать его в мою сеть LSTM?

Нужно ли делать y_train длиной 54837488? 

model = Sequential()
model.add(LSTM(32, input_shape=(X_train.shape[1], X_train.shape[2])))
model.add(Dense(32, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(y_train.shape[1], activation='softmax'))
...