Выбор time_step для LSTM

Question

Я пытаюсь изменить свой ввод для моей сети LSTM. У меня есть тренировочные данные train_x (20214000 столбцов x 9 строк) и train_y (20214000 столбцов x 1 строка).

Как мне изменить форму train_x так, чтобы я мог подать его в мой RNN?

У меня 9 функций, поэтому это будет что-то вроде:

train_x.reshape(?,?,9) и train_y.reshape(?,?,1)

Answer 1

Массивы, введенные в LSTM, имеют форму: (N_SAMPLES, SEQUENCE_LENGTH, N_FEATURES).