当前位置: 代码迷 >> python >> keras GRU不会采用我的简单二维数组
  详细解决方案

keras GRU不会采用我的简单二维数组

热度:57   发布时间:2023-07-14 08:57:42.0

我创建了一个玩具数据集,该数据集具有(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)

`

LSTM / GRU需要一个3D数组作为输入。您需要将数据转换为带有3Dimension的数组,即(样本,时间步长,特征)。 您可以使用numpy的reshape()函数执行转换。 请参阅

  相关解决方案