【深度学习】多层感知器MLP模型对 MNIST 数据集中的手写数字进行分类
变量,用于累计所有批次的损失之和,这里的train_loss是所有批次的损失之和,所以计算全部样本的平均损失时需要除以总批次数,correct是预测正确的样本总数,计算整个epoch总体正确率,需要除以样本总数量。,本质上就是对全部输入加权求和,它要求输入数据集的形状为一维的,如果使用批量运算,则增加一个batch维,也就是需要输入数据是二维的形状,第一维是batch维,第二维是数据特征,即(batch_size,feature_length)形式。最后输出 “Done!,以及模型和优化器的初始化。
原创
2024-05-23 21:22:49 ·
581 阅读 ·
0 评论