问题描述
我创建了一个玩具数据集,该数据集具有(n_rows,51)个50个特征和1个真值的暗淡。 当我做一个简单的线性回归时,keras起作用。 但是,一旦尝试使用GRU或LSTM,就不会起作用。 以下是尝试使自己轻松进行的尝试。 谁能帮我找出问题所在。
`
model = Sequential()
model.add(GRU(50))
model.add(Flatten())
model.add(Reshape((None, 50)))
model.add(Dense(units=1, activation='softmax'))
model.compile(loss='mean_squared_error', optimizer='adam', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=10, batch_size=20)
`
1楼
LSTM / GRU需要一个3D数组作为输入。您需要将数据转换为带有3Dimension的数组,即(样本,时间步长,特征)。 您可以使用numpy的reshape()函数执行转换。 请参阅