首页 > 解决方案 > 将数组数组输入循环神经网络(Keras)

问题描述

我目前有表格中的坐标数据

coordinate = [x,y,z,t,dx,dy,dz,dt]

并且这些坐标组形成一条看起来像这样的路径

path = [[coordinate 1], [coordinate2], ...]

最终我的所有数据都存储为单个数组,就像这样(数组数组的数组)

data = [path1,path2,path3 ....]

我想在 Keras 中输入一条路径,特别是一个 LSTM 来预测该路径的下一个坐标。我怎样才能做到这一点?我知道时间步长是 1,批量大小可以变化为无,但 Keras 会抛出形状输入错误。

目前我的基本模型是这个(我知道选项可能不正确)

model.add(LSTM(units=1,return_sequences=True,input_shape=(None,8)))
model.add(Dropout(0.2))

#model.add(LSTM(units=1,return_sequences=True,input_shape=(None,8)))                                                                                                                                               
#model.add(Dropout(0.2))                                                                                                                                                                                           

model.add(Dense(1))

model.compile(optimizer='adam',loss='mean_squared_error')

谢谢

标签: pythonarraystensorflowkerasrecurrent-neural-network

解决方案


推荐阅读