CNN + LSTM МОДЕЛЬ - PullRequest
Новая
       40

CNN + LSTM МОДЕЛЬ

0 голосов
/ 01 октября 2019 
train input shape :  (13974, 100, 6, 5) 
train output shape :   (13974, 1, 6, 5) 
test input shape :   (3494, 100, 6, 5) 
test output shape :   (3494, 1, 6, 5)

13974 - последовательности 100 - отметка времени 6 - местоположения 5 - переменные

Мне нужна моя модель для прогнозирования следующей 101-й отметки времени для 6 местоположений и 5 переменных для каждого местоположения. 

model = Sequential()
model.add(TimeDistributed(Conv2D(32, 5, 5,input_shape=(100,6,5,1))))
model.add(TimeDistributed(MaxPooling2D(pool_size=(4, 4))))
model.add(Activation('relu'))
model.add(TimeDistributed(Flatten()))
model.add(LSTM(100, return_sequences=True))
model.add(TimeDistributed(Dense(5)))  
model.compile(loss='mae', optimizer='adam', metrics=['accuracy'])

это моя модель. Я получил ошибку ValueError: шаг должен быть длиной 1, 1 или 3, но был 2

Мне нужно скормить поезд input : (13974, 100, 6, 5) и предсказатьпоезд utput:(13974, 1, 6, 5). Пожалуйста, исправьте мою модель и внесите необходимые исправления?

...