通过极简方案构建手写数字识别模型

上一节介绍了创新性的“横纵式”教学法，有助于深度学习初学者快速掌握深度学习理论知识，并在过程中让读者获得到真实建模的实战体验。在“横纵式”教学法中，纵向概要介绍模型的基本代码结构和极简实现方案，如 图1 所示。本节将使用这种极简实现方案快速完成手写数字识别的建模。

通过极简方案快速构建手写数字识别模型 - 图1

图1：“横纵式”教学法—纵向极简实现方案

前提条件

在数据处理前，首先要加载飞桨平台与“手写数字识别”模型相关类库，代码如下。

  1. #加载飞桨和相关类库
  2. import paddle
  3. import paddle.fluid as fluid
  4. from paddle.fluid.dygraph.nn import FC
  5. import numpy as np
  6. import os
  7. from PIL import Image

数据处理

飞桨提供了多个封装好的数据集API，涵盖计算机视觉、自然语言处理、推荐系统等多个领域，帮助读者快速完成机器学习任务。如在手写数字识别模型中，通过paddle.dataset.mnist.train()可以直接获取处理好的MNIST训练集、验证集和测试集，飞桨API支持如下常见的学术数据集：

  • mnist
  • cifar
  • Conll05
  • imdb
  • imikolov
  • movielens
  • sentiment
  • uci_housing
  • wmt14
  • wmt16

通过paddle.dataset.mnist.train()函数设置数据读取器，batch_size设置为8，即一个批次有8张图片和8个标签，代码如下所示。

  1. # 如果～/.cache/paddle/dataset/mnist/目录下没有MNIST数据，API会自动将MINST数据下载到该文件夹下
  2. # 设置数据读取器，读取MNIST数据训练集
  3. trainset = paddle.dataset.mnist.train()
  4. # 包装数据读取器，每次读取的数据数量设置为batch_size=8
  5. train_reader = paddle.batch(trainset, batch_size=8)
  1. [==================================================]t/train-images-idx3-ubyte.gz not found, downloading https://dataset.bj.bcebos.com/mnist/train-images-idx3-ubyte.gz
  2. [==================================================]t/train-labels-idx1-ubyte.gz not found, downloading https://dataset.bj.bcebos.com/mnist/train-labels-idx1-ubyte.gz

paddle.batch函数将MNIST数据集拆分成多个批次，通过如下代码读取第一个批次的数据内容，观察数据打印结果。

  1. # 以迭代的形式读取数据
  2. for batch_id, data in enumerate(train_reader()):
  3.     # 获得图像数据，并转为float32类型的数组
  4.     img_data = np.array([x[0] for x in data]).astype('float32')
  5.     # 获得图像标签数据，并转为float32类型的数组
  6.     label_data = np.array([x[1] for x in data]).astype('float32')
  7.     # 打印数据形状
  8.     print("图像数据形状和对应数据为:", img_data.shape, img_data[0])
  9.     print("图像标签形状和对应数据为:", label_data.shape, label_data[0])
  10.     break
  12. print("\n打印第一个batch的第一个图像，对应标签数字为{}".format(label_data[0]))
  13. # 显示第一batch的第一个图像
  14. import matplotlib.pyplot as plt
  15. img = np.array(img_data[0]+1)*127.5
  16. img = np.reshape(img, [28, 28]).astype(np.uint8)
  18. plt.figure("Image") # 图像窗口名称
  19. plt.imshow(img)
  20. plt.axis('on') # 关掉坐标轴为 off
  21. plt.title('image') # 图像题目
  22. plt.show()
  1. 图像数据形状和对应数据为: (8, 784) [-1.         -1.         -1.         -1.         -1.         -1.
  2.  -1.         -1.         -1.         -1.         -1.         -1.
  3.  -1.         -1.         -1.         -1.         -1.         -1.
  4.  -1.         -1.         -1.         -1.         -1.         -1.
  5.  -1.         -1.         -1.         -1.         -1.         -1.
  6.  -1.         -1.         -1.         -1.         -1.         -1.
  7.  -1.         -1.         -1.         -1.         -1.         -1.
  8.  -1.         -1.         -1.         -1.         -1.         -1.
  9.  -1.         -1.         -1.         -1.         -1.         -1.
  10.  -1.         -1.         -1.         -1.         -1.         -1.
  11.  -1.         -1.         -1.         -1.         -1.         -1.
  12.  -1.         -1.         -1.         -1.         -1.         -1.
  13.  -1.         -1.         -1.         -1.         -1.         -1.
  14.  -1.         -1.         -1.         -1.         -1.         -1.
  15.  -1.         -1.         -1.         -1.         -1.         -1.
  16.  -1.         -1.         -1.         -1.         -1.         -1.
  17.  -1.         -1.         -1.         -1.         -1.         -1.
  18.  -1.         -1.         -1.         -1.         -1.         -1.
  19.  -1.         -1.         -1.         -1.         -1.         -1.
  20.  -1.         -1.         -1.         -1.         -1.         -1.
  21.  -1.         -1.         -1.         -1.         -1.         -1.
  22.  -1.         -1.         -1.         -1.         -1.         -1.
  23.  -1.         -1.         -1.         -1.         -1.         -1.
  24.  -1.         -1.         -1.         -1.         -1.         -1.
  25.  -1.         -1.         -1.         -1.         -1.         -1.
  26.  -1.         -1.         -0.9764706  -0.85882354 -0.85882354 -0.85882354
  27.  -0.01176471  0.06666672  0.37254906 -0.79607844  0.30196083  1.
  28.   0.9372549  -0.00392157 -1.         -1.         -1.         -1.
  29.  -1.         -1.         -1.         -1.         -1.         -1.
  30.  -1.         -1.         -0.7647059  -0.7176471  -0.26274508  0.20784318
  31.   0.33333337  0.9843137   0.9843137   0.9843137   0.9843137   0.9843137
  32.   0.7647059   0.34901965  0.9843137   0.8980392   0.5294118  -0.4980392
  33.  -1.         -1.         -1.         -1.         -1.         -1.
  34.  -1.         -1.         -1.         -1.         -1.         -0.6156863
  35.   0.8666667   0.9843137   0.9843137   0.9843137   0.9843137   0.9843137
  36.   0.9843137   0.9843137   0.9843137   0.96862745 -0.27058822 -0.35686272
  37.  -0.35686272 -0.56078434 -0.69411767 -1.         -1.         -1.
  38.  -1.         -1.         -1.         -1.         -1.         -1.
  39.  -1.         -1.         -1.         -0.85882354  0.7176471   0.9843137
  40.   0.9843137   0.9843137   0.9843137   0.9843137   0.5529412   0.427451
  41.   0.9372549   0.8901961  -1.         -1.         -1.         -1.
  42.  -1.         -1.         -1.         -1.         -1.         -1.
  43.  -1.         -1.         -1.         -1.         -1.         -1.
  44.  -1.         -1.         -0.372549    0.22352946 -0.1607843   0.9843137
  45.   0.9843137   0.60784316 -0.9137255  -1.         -0.6627451   0.20784318
  46.  -1.         -1.         -1.         -1.         -1.         -1.
  47.  -1.         -1.         -1.         -1.         -1.         -1.
  48.  -1.         -1.         -1.         -1.         -1.         -1.
  49.  -1.         -0.8901961  -0.99215686  0.20784318  0.9843137  -0.29411763
  50.  -1.         -1.         -1.         -1.         -1.         -1.
  51.  -1.         -1.         -1.         -1.         -1.         -1.
  52.  -1.         -1.         -1.         -1.         -1.         -1.
  53.  -1.         -1.         -1.         -1.         -1.         -1.
  54.  -1.          0.09019613  0.9843137   0.4901961  -0.9843137  -1.
  55.  -1.         -1.         -1.         -1.         -1.         -1.
  56.  -1.         -1.         -1.         -1.         -1.         -1.
  57.  -1.         -1.         -1.         -1.         -1.         -1.
  58.  -1.         -1.         -1.         -1.         -1.         -0.9137255
  59.   0.4901961   0.9843137  -0.45098037 -1.         -1.         -1.
  60.  -1.         -1.         -1.         -1.         -1.         -1.
  61.  -1.         -1.         -1.         -1.         -1.         -1.
  62.  -1.         -1.         -1.         -1.         -1.         -1.
  63.  -1.         -1.         -1.         -1.         -0.7254902   0.8901961
  64.   0.7647059   0.254902   -0.15294117 -0.99215686 -1.         -1.
  65.  -1.         -1.         -1.         -1.         -1.         -1.
  66.  -1.         -1.         -1.         -1.         -1.         -1.
  67.  -1.         -1.         -1.         -1.         -1.         -1.
  68.  -1.         -1.         -1.         -0.36470586  0.88235295  0.9843137
  69.   0.9843137  -0.06666666 -0.8039216  -1.         -1.         -1.
  70.  -1.         -1.         -1.         -1.         -1.         -1.
  71.  -1.         -1.         -1.         -1.         -1.         -1.
  72.  -1.         -1.         -1.         -1.         -1.         -1.
  73.  -1.         -1.         -0.64705884  0.45882356  0.9843137   0.9843137
  74.   0.17647064 -0.7882353  -1.         -1.         -1.         -1.
  75.  -1.         -1.         -1.         -1.         -1.         -1.
  76.  -1.         -1.         -1.         -1.         -1.         -1.
  77.  -1.         -1.         -1.         -1.         -1.         -1.
  78.  -1.         -0.8745098  -0.27058822  0.9764706   0.9843137   0.4666667
  79.  -1.         -1.         -1.         -1.         -1.         -1.
  80.  -1.         -1.         -1.         -1.         -1.         -1.
  81.  -1.         -1.         -1.         -1.         -1.         -1.
  82.  -1.         -1.         -1.         -1.         -1.         -1.
  83.  -1.          0.9529412   0.9843137   0.9529412  -0.4980392  -1.
  84.  -1.         -1.         -1.         -1.         -1.         -1.
  85.  -1.         -1.         -1.         -1.         -1.         -1.
  86.  -1.         -1.         -1.         -1.         -1.         -1.
  87.  -1.         -1.         -0.6392157   0.0196079   0.43529415  0.9843137
  88.   0.9843137   0.62352943 -0.9843137  -1.         -1.         -1.
  89.  -1.         -1.         -1.         -1.         -1.         -1.
  90.  -1.         -1.         -1.         -1.         -1.         -1.
  91.  -1.         -1.         -1.         -1.         -0.69411767  0.16078436
  92.   0.79607844  0.9843137   0.9843137   0.9843137   0.9607843   0.427451
  93.  -1.         -1.         -1.         -1.         -1.         -1.
  94.  -1.         -1.         -1.         -1.         -1.         -1.
  95.  -1.         -1.         -1.         -1.         -1.         -1.
  96.  -0.8117647  -0.10588235  0.73333335  0.9843137   0.9843137   0.9843137
  97.   0.9843137   0.5764706  -0.38823527 -1.         -1.         -1.
  98.  -1.         -1.         -1.         -1.         -1.         -1.
  99.  -1.         -1.         -1.         -1.         -1.         -1.
  100.  -1.         -1.         -0.81960785 -0.4823529   0.67058825  0.9843137
  101.   0.9843137   0.9843137   0.9843137   0.5529412  -0.36470586 -0.9843137
  102.  -1.         -1.         -1.         -1.         -1.         -1.
  103.  -1.         -1.         -1.         -1.         -1.         -1.
  104.  -1.         -1.         -1.         -1.         -0.85882354  0.3411765
  105.   0.7176471   0.9843137   0.9843137   0.9843137   0.9843137   0.5294118
  106.  -0.372549   -0.92941177 -1.         -1.         -1.         -1.
  107.  -1.         -1.         -1.         -1.         -1.         -1.
  108.  -1.         -1.         -1.         -1.         -1.         -1.
  109.  -0.5686275   0.34901965  0.77254903  0.9843137   0.9843137   0.9843137
  110.   0.9843137   0.9137255   0.04313731 -0.9137255  -1.         -1.
  111.  -1.         -1.         -1.         -1.         -1.         -1.
  112.  -1.         -1.         -1.         -1.         -1.         -1.
  113.  -1.         -1.         -1.         -1.          0.06666672  0.9843137
  114.   0.9843137   0.9843137   0.6627451   0.05882359  0.03529418 -0.8745098
  115.  -1.         -1.         -1.         -1.         -1.         -1.
  116.  -1.         -1.         -1.         -1.         -1.         -1.
  117.  -1.         -1.         -1.         -1.         -1.         -1.
  118.  -1.         -1.         -1.         -1.         -1.         -1.
  119.  -1.         -1.         -1.         -1.         -1.         -1.
  120.  -1.         -1.         -1.         -1.         -1.         -1.
  121.  -1.         -1.         -1.         -1.         -1.         -1.
  122.  -1.         -1.         -1.         -1.         -1.         -1.
  123.  -1.         -1.         -1.         -1.         -1.         -1.
  124.  -1.         -1.         -1.         -1.         -1.         -1.
  125.  -1.         -1.         -1.         -1.         -1.         -1.
  126.  -1.         -1.         -1.         -1.         -1.         -1.
  127.  -1.         -1.         -1.         -1.         -1.         -1.
  128.  -1.         -1.         -1.         -1.         -1.         -1.
  129.  -1.         -1.         -1.         -1.         -1.         -1.
  130.  -1.         -1.         -1.         -1.         -1.         -1.
  131.  -1.         -1.         -1.         -1.        ]
  132. 图像标签形状和对应数据为: (8,) 5.0
  133.  
  134. 打印第一个batch的第一个图像，对应标签数字为5.0
  1. <Figure size 640x480 with 1 Axes>

从打印结果看，从数据加载器train_loader()中读取一次数据，可以得到形状为（8, 784）的图像数据和形状为（8,）的标签数据。其中，形状中的数字8与设置的batch_size大小对应，784为MINIST数据集中每个图像的像素大小(28*28)。

此外，从打印的图像数据来看，图像数据的范围是[-1, 1]，表明这是已经完成图像归一化后的图像数据，并且空白背景部分的值是-1。将图像数据反归一化，并使用matplotlib工具包将其显示出来，如 图2 所示。图片显示的数字是5，和对应标签数字一致。

通过极简方案快速构建手写数字识别模型 - 图2

图2：matplotlib打印结果示意图

说明：

飞桨将维度是2828的手写数字数据图像转成向量形式存储，因此使用飞桨数据读取到的手写数字图像是长度为784（2828）的向量。

飞桨API的使用方法

熟练掌握飞桨API的使用方法，是使用飞桨构建各类深度学习任务的基础，也是开发者必须掌握的技能，下面介绍飞桨API获取方式和使用方法。

1. 飞桨API文档获取方式

登录“飞桨官网->文档->API Reference”，获取飞桨API文档，如 图3 所示。

通过极简方案快速构建手写数字识别模型 - 图3

图3：飞桨API文档

2. 通过搜索和分类浏览两种方式查阅API文档

如果用户知道需要查阅的API名称，可通过页面右上角的搜索框，快速获取API。

如果想全面了解飞桨API文档内容，也可以在API Reference首页，单击“API功能分类”，通过概念分类获取不同职能的API，如 图4 所示

通过极简方案快速构建手写数字识别模型 - 图4

图4：飞桨API功能分类页面

在API功能分类的页面，用户可以根据神经网络建模的逻辑概念来浏览相应部分的API，如优化器、网络层、评价指标、模型保存和加载等。

3. API文档使用方法

飞桨每个API的文档结构一致，包含接口形式、功能说明和计算公式、参数和返回值、代码示例四个部分。 以abs函数为例，API文档结构如 图5 所示。通过飞桨API文档，读者不仅可以详细查看函数功能，还可以通过可运行的代码示例来实践API的使用。

通过极简方案快速构建手写数字识别模型 - 图5

图5：abs函数的API文档

模型设计

在房价预测深度学习任务中，我们使用了单层且没有非线性变换的模型，取得了理想的预测效果。在手写数字识别中，我们依然使用这个模型预测输入的图形数字值。其中，模型的输入为784维（28*28）数据，输出为1维数据，如 图6 所示。

通过极简方案快速构建手写数字识别模型 - 图6

图6：手写数字识别网络模型

输入像素的位置排布信息对理解图像内容非常重要（如将原始尺寸为2828图像的像素按照7112的尺寸排布，那么其中的数字将不可识别），因此网络的输入设计为2828的尺寸，而不是1784，以便于模型能够正确处理像素之间的空间信息。

说明：

事实上，采用只有一层的简单网络（对输入求加权和）时并没有处理位置关系信息，因此可以猜测出此模型的预测效果可能有限。在后续优化环节介绍的卷积神经网络则更好的考虑了这种位置关系信息，模型的预测效果也会有显著提升。

下面以类的方式组建手写数字识别的网络，代码如下所示。

  1. # 定义mnist数据识别网络结构，同房价预测网络
  2. class MNIST(fluid.dygraph.Layer):
  3.     def __init__(self, name_scope):
  4.         super(MNIST, self).__init__(name_scope)
  5.         name_scope = self.full_name()
  6.         # 定义一层全连接层，输出维度是1，激活函数为None，即不使用激活函数
  7.         self.fc = FC(name_scope, size=1, act=None)
  9.     # 定义网络结构的前向计算过程
  10.     def forward(self, inputs):
  11.         outputs = self.fc(inputs)
  12.         return outputs

训练配置

训练配置需要先生成模型实例（设为“训练”状态），再设置优化算法和学习率（使用随机梯度下降SGD，学习率设置为0.01），代码如下所示。

  1. # 定义飞桨动态图工作环境
  2. with fluid.dygraph.guard():
  3.     # 声明网络结构
  4.     model = MNIST("mnist")
  5.     # 启动训练模式
  6.     model.train()
  7.     # 定义数据读取函数，数据读取batch_size设置为16
  8.     train_loader = paddle.batch(paddle.dataset.mnist.train(), batch_size=16)
  9.     # 定义优化器，使用随机梯度下降SGD优化器，学习率设置为0.001
  10.     optimizer = fluid.optimizer.SGDOptimizer(learning_rate=0.001)

训练过程

训练过程采用二层循环嵌套方式，训练完成后需要保存模型参数，以便后续使用。

  • 内层循环：负责整个数据集的一次遍历，遍历数据集采用分批次（batch）方式。
  • 外层循环：定义遍历数据集的次数，本次训练中外层循环10次，通过参数EPOCH_NUM设置。 
  1. # 通过with语句创建一个dygraph运行的context，
  2. # 动态图下的一些操作需要在guard下进行
  3. with fluid.dygraph.guard():
  4.     model = MNIST("mnist")
  5.     model.train()
  6.     train_loader = paddle.batch(paddle.dataset.mnist.train(), batch_size=16)
  7.     optimizer = fluid.optimizer.SGDOptimizer(learning_rate=0.001)
  8.     EPOCH_NUM = 10
  9.     for epoch_id in range(EPOCH_NUM):
  10.         for batch_id, data in enumerate(train_loader()):
  11.             #准备数据，格式需要转换成符合框架要求的
  12.             image_data = np.array([x[0] for x in data]).astype('float32')
  13.             label_data = np.array([x[1] for x in data]).astype('float32').reshape(-1, 1)
  14.             # 将数据转为飞桨动态图格式
  15.             image = fluid.dygraph.to_variable(image_data)
  16.             label = fluid.dygraph.to_variable(label_data)
  18.             #前向计算的过程
  19.             predict = model(image)
  21.             #计算损失，取一个批次样本损失的平均值
  22.             loss = fluid.layers.square_error_cost(predict, label)
  23.             avg_loss = fluid.layers.mean(loss)
  25.             #每训练了1000批次的数据，打印下当前Loss的情况
  26.             if batch_id !=0 and batch_id  % 1000 == 0:
  27.                 print("epoch: {}, batch: {}, loss is: {}".format(epoch_id, batch_id, avg_loss.numpy()))
  29.             #后向传播，更新参数的过程
  30.             avg_loss.backward()
  31.             optimizer.minimize(avg_loss)
  32.             model.clear_gradients()
  34.     # 保存模型
  35.     fluid.save_dygraph(model.state_dict(), 'mnist')
  1. epoch: 0, batch: 1000, loss is: [2.2036428]
  2. epoch: 0, batch: 2000, loss is: [4.1068788]
  3. epoch: 0, batch: 3000, loss is: [3.514595]
  4. epoch: 1, batch: 1000, loss is: [2.0001]
  5. epoch: 1, batch: 2000, loss is: [3.900276]
  6. epoch: 1, batch: 3000, loss is: [3.37518]
  7. epoch: 2, batch: 1000, loss is: [1.9623451]
  8. epoch: 2, batch: 2000, loss is: [3.7900715]
  9. epoch: 2, batch: 3000, loss is: [3.2977998]
  10. epoch: 3, batch: 1000, loss is: [1.9503691]
  11. epoch: 3, batch: 2000, loss is: [3.7407346]
  12. epoch: 3, batch: 3000, loss is: [3.2407851]
  13. epoch: 4, batch: 1000, loss is: [1.9477944]
  14. epoch: 4, batch: 2000, loss is: [3.7205358]
  15. epoch: 4, batch: 3000, loss is: [3.1938488]
  16. epoch: 5, batch: 1000, loss is: [1.9492074]
  17. epoch: 5, batch: 2000, loss is: [3.7143831]
  18. epoch: 5, batch: 3000, loss is: [3.1537359]
  19. epoch: 6, batch: 1000, loss is: [1.9525008]
  20. epoch: 6, batch: 2000, loss is: [3.7151768]
  21. epoch: 6, batch: 3000, loss is: [3.1190662]
  22. epoch: 7, batch: 1000, loss is: [1.9567086]
  23. epoch: 7, batch: 2000, loss is: [3.7194653]
  24. epoch: 7, batch: 3000, loss is: [3.0890307]
  25. epoch: 8, batch: 1000, loss is: [1.9613326]
  26. epoch: 8, batch: 2000, loss is: [3.7254934]
  27. epoch: 8, batch: 3000, loss is: [3.063015]
  28. epoch: 9, batch: 1000, loss is: [1.96609]
  29. epoch: 9, batch: 2000, loss is: [3.7323272]
  30. epoch: 9, batch: 3000, loss is: [3.040489]

通过观察上述代码可以发现，手写数字识别的代码几乎与房价预测任务一致，如果不是下述读取数据的两行代码有所差异，我们会误认为这是房价预测的模型。

  1.    #准备数据，格式需要转换成符合框架要求的
  2.    image_data = np.array([x[0] for x in data]).astype('float32')
  3.    label_data = np.array([x[1] for x in data]).astype('float32').reshape(-1, 1)

另外，从训练过程中损失所发生的变化可以发现，虽然损失整体上在降低，但到训练的最后一轮，损失函数值依然较高。可以猜测手写数字识别完全复用房价预测的代码，训练效果并不好。接下来我们通过模型测试，获取模型训练的真实效果。

模型测试

模型测试的主要目的是验证训练好的模型是否能正确识别出数字，包括如下四步：

  • 声明实例
  • 加载模型：加载训练过程中保存的模型参数，
  • 灌入数据：将测试样本传入模型，模型的状态设置为校验状态（eval），显式告诉框架我们接下来只会使用前向计算的流程，不会计算梯度和梯度反向传播。
  • 获取预测结果，取整后作为预测标签输出。

在模型测试之前，需要先从’./demo/example_0.jpg’文件中读取样例图片，并进行归一化处理。

  1. # 导入图像读取第三方库
  2. import matplotlib.image as mpimg
  3. import matplotlib.pyplot as plt
  4. # 读取图像
  5. example = mpimg.imread('./work/example_0.png')
  6. # 显示图像
  7. plt.imshow(example)
  1. <matplotlib.image.AxesImage at 0x7feb1365cd10>

通过极简方案快速构建手写数字识别模型 - 图7

  1. <Figure size 432x288 with 1 Axes>
  1. # 读取一张本地的样例图片，转变成模型输入的格式
  2. def load_image(img_path):
  3.     # 从img_path中读取图像，并转为灰度图
  4.     im = Image.open(img_path).convert('L')
  5.     print(np.array(im))
  6.     im = im.resize((28, 28), Image.ANTIALIAS)
  7.     im = np.array(im).reshape(1, -1).astype(np.float32)
  8.     # 图像归一化，保持和数据集的数据范围一致
  9.     im = 1 - im / 127.5
  10.     return im
  12. # 定义预测过程
  13. with fluid.dygraph.guard():
  14.     model = MNIST("mnist")
  15.     params_file_path = 'mnist'
  16.     img_path = './work/example_0.png'
  17. # 加载模型参数
  18.     model_dict, _ = fluid.load_dygraph("mnist")
  19.     model.load_dict(model_dict)
  20. # 灌入数据
  21.     model.eval()
  22.     tensor_img = load_image(img_path)
  23.     result = model(fluid.dygraph.to_variable(tensor_img))
  24. #  预测输出取整，即为预测的数字，打印结果
  25.     print("本次预测的数字是", result.numpy().astype('int32'))
  1. [[255 255 255 ... 255 255 255]
  2.  [255 255 255 ... 255 255 255]
  3.  [255 255 255 ... 255 255 255]
  4.  ...
  5.  [255 255 255 ... 255 255 255]
  6.  [255 255 255 ... 255 255 255]
  7.  [255 255 255 ... 255 255 255]]
  8. 本次预测的数字是 [[4]]

从打印结果来看，模型预测出的数字是与实际输出的图片的数字不一致。这里只是验证了一个样本的情况，如果我们尝试更多的样本，可发现许多数字图片识别结果是错误的。因此完全复用房价预测的实验并不适用于手写数字识别任务！

接下来我们会对手写数字识别实验模型进行逐一改进，直到获得令人满意的结果。

作业 2-1：

  • 使用飞桨API paddle.dataset.mnist的test函数获得测试集数据，计算当前模型的准确率。

  • 怎样进一步提高模型的准确率？可以再接下来内容开始前，写出你想到的优化思路。