机器学习系统:设计和实现
上QQ阅读APP看书,第一时间看更新

2.2.6 测试和验证

测试是将测试数据集输入到模型,运行得到输出的过程,通常在训练过程中,每训练一定的数据量后就会测试一次,以验证模型的泛化能力。MindSpore使用model.eval接口读入测试数据集,如代码2.9所示。

代码2.9 定义模型验证

在训练完毕后,参数保存在检查点(checkpoint)中,可以将训练好的参数加载到模型中进行验证,如代码2.10所示。

代码2.10 模型验证