【发布时间】:2022-08-19 19:55:22
【问题描述】:
我有一些像素运动数据,它有 5 个特征和 3715489 个训练样本。我不断收到这个错误,我不知道我应该为 LSTM 制作什么 input_shape。
X_train 形状为 (3715489,5)。我需要重塑这个吗?
y_train 形状为 (3715489, 8)
这是我的代码:
model = Sequential()
model.add(LSTM(256,return_sequences=True, input_shape=(5,)))
model.add(Dense(8, activation=\'sigmoid\'))
model.compile(optimizer=\'adam\',loss=\'categorical_crossentropy\', metrics=[\'accuracy\'])
print(model.summary())
model.fit(x_train, y_train, epochs=100,batch_size=320)
-
请编辑问题以将其限制为具有足够详细信息的特定问题,以确定适当的答案。
标签: tensorflow machine-learning keras lstm recurrent-neural-network