【MXNet学习24】4.3 测试代码初探

在训练得到一个模型后，一般可以通过一个测试代码测试模型在指定输入图像时的输出结果，从而判断训练得到的模型是否有效。有时候测试代码也被称为demo，可以用来展示模型训练的效果，因此测试代码是一个完整项目必不可少的部分。

完整的测试代码如代码清单4-2所示，主要包含模型导入、数据读取和预测输出3个部分，下面假设这些代码都保存在test_mnist_code4-2.py脚本中。

代码清单4-2 MNIST手写数字体分类测试代码

import mxnet as mx
import numpy as np

def load_model(model_prefix, index, context, data_shapes, label_shapes):
    sym, arg_params