输入样本train_x1 标签train_y1
样本与标签都是(20000,10)
然后我reshape变成三维(20000,1,10)报错 求该如何修改格式
输入十个数值,输出十个数值。
train_x1 = np.reshape(train_x1, (train_x1.shape[0],1,train_x1.shape[1]))
train_y1 = np.reshape(train_y1, (train_y1.shape[0],1,train_y1.shape[1]))
model = Sequential()
model.add(LSTM(50, input_shape=(train_x1.shape[1], train_x1.shape[2])))
model.add(Dense(10))
model.compile(loss='mse', optimizer='adam')
model.fit(train_x1, train_y1, nb_epoch = 300, batch_size = 10)
model.save_weights('LSTM.model')