LSTM应用于MNIST数据集分类(与CNN做对比)

LSTM应用于MNIST数据集分类

• 1、概述
• 2、LSTM实现
• • 2.1 载入数据集
  • 2.2 创建模型
  • 2.3 定义优化器
  • 2.4 编译模型
  • 2.5 训练模型
  • 2.6 打印模型摘要
  • 2.7 绘制acc和loss曲线
• 3、CNN实现

1、概述

  LSTM网络是序列模型，一般比较适合处理序列问题。这里把它用于手写数字图片的分类，其实就相当于把图片看作序列。

  一张MNIST数据集的图片是 28 × 28 28\times 28