【发布时间】:2021-11-26 23:40:57
【问题描述】:
我有一个数据集,其中每个样本都是可变数量的时间步长,并且在每个时间步长内有准确的 5 个浮点数。我想训练一个预测单个值的 RNN/LSTM:
sample_1 = [[1.0, 2.0, 3.0, 4.0, 5.0], [5.0, 4.0, 3.0, 2.0, 1.0].....]
train_x = [sample_1, sample_2.....] train_y = [123.0, 456.0.....]
我的 Sequential Keras LSTM 模型如下所示:
model = Sequential()
model.add(LSTM(units=4, dropout=0.5))
model.add(Dense(units=4, activation='relu'))
model.add(Dense(1))
model.compile(loss='mean_absolute_error',
optimizer=Adam())
训练:
_ = model.fit(
train_x,
train_y,
validation_data=(validation_x, validation_y),
shuffle=True,
batch_size=16,
epochs=10,
verbose=2)
但是,我立即收到以下错误:
“ValueError:无法将 NumPy 数组转换为张量(不支持的对象类型列表)。”
我的模型设置有什么明显错误吗?我可以简单地使用列表的嵌套列表作为输入示例吗?
【问题讨论】:
标签: python tensorflow machine-learning keras lstm