【发布时间】:2017-11-10 17:34:22
【问题描述】:
我有一个5000 by 9 2d numpy 特征数组trainX,它们是时间序列的特征。我还有一个 1d numpy 浮点特征标签数组trainY。例如,这正是 scikit-learn 所需的格式。
我想将这些与 keras+LSTM 一起使用。这是我目前的代码:
NUM_EPOCHS = 20
model = Sequential()
model.add(LSTM(8, input_shape=(1, window_size)))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(trainX, trainY, epochs=NUM_EPOCHS, batch_size=1, verbose=2)
但这不起作用,因为 keras 需要 trainX 以不同的格式。我已经阅读了手册,但我无法理解这到底是什么。
如何将我的数据转换成 keras 可以接受的格式?
【问题讨论】:
标签: python machine-learning tensorflow keras