机器学习系统与优化 教案 案例12-第四章 FNN用于MNIST手写数字识别案例.docx

案例12——FNN用于MNIST手写数字识别案例

利用单隐藏层前馈神经网络（FNN）用于MNIST手写数字识别任务。

数据集

选用MNIST数据集进行模型的训练与测试。MNIST数据集是一个经典的用于图像识别的数据集，其中包含了0-9的手写数字灰度图像，共计60000张训练图像和10000张测试图像，每张图像的大小为28×28像素。MNIST数据集可从官网/exdb/mnist/进行下载。

代码

importtorch

importtorch.nnasnn

importtorch.optimasoptim

fromtorchvisionimportdatas