基于pytorch的手写数字识别(超高准确率!)
10轮后,在训练集上的识别准确率达到99.97%,而损失值为0.0009,在测试集上的准确率达到99.54%,损失值为0.0213,其中训练集和测试集上的损失值准确率如下图所示。下表为显示的混淆矩阵值,可以看出极大部分的数字都有被准确率测,只有少数部分的数字没有被准确预测,通过分析矩阵中的数据,可以定位模型的误差并针对性改进。:使用_get_conv_output_shape()函数通过假输入计算卷积层输出特征图展平后的大小,为全连接层做准备。下表为训练集和测试集上的损失值和识别准确率的输入结果。
原创
2025-01-02 16:23:29 ·
1563 阅读 ·
0 评论