如何为二维数据构建LSTM网络?

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何为二维数据构建LSTM网络?相关的知识,希望对你有一定的参考价值。

我看到Keras中的LSTM接受(batch_size,timesteps,data_dim)作为输入形状。但我想知道是否有一种简单的方法可以将这些扩展到4维度?例如,我们如何将lstm应用于我们拥有64 x 64图像的移动MNIST数据集?

答案

您可以使用CNN模型在将图像输入LSTM之前提取图像的特征。

# define CNN model
cnn = Sequential()
cnn.add(Conv2D(...))
cnn.add(MaxPooling2D(...))
cnn.add(Flatten())

# define LSTM model
model = Sequential()
model.add(TimeDistributed(cnn, ...))
model.add(LSTM(..))
model.add(Dense(...))

以上是关于如何为二维数据构建LSTM网络?的主要内容,如果未能解决你的问题,请参考以下文章

如何为 LSTM 实现 Keras 自定义损失函数

如何为 XSLT 代码片段配置 CruiseControl 的 C# 版本?

如何为 keras 中的 LSTM 回归准备输入数据?

如何为 LSTM 重塑数据 - 时间序列多类分类

如何为多对一二元分类 LSTM 准备数据?

如何为序列项目训练 LSTM 模型?