Mxnet MNIST 训练示例 returns 几乎不变的 rmse

Mxnet MNIST training example returns almost constant rmse

我通过添加 rmse 损失对 fit.py mxnet python 示例文件(图像分类)做了一点修改:

# evaluation metrices
eval_metrics = ['accuracy']
eval_metrics.append('rmse') 

然后运行MNIST训练样例,可以观察到rmse一直在5.2左右,准确率上升到99%左右。

我们不应该观察到 RMSE 下降吗?

非常感谢 阿尔

均方根误差 (RMSE) 是用于回归问题的指标。在回归问题中,网络正在预测一个实数,并且该预测的质量可以作为预测值与期望值之间的数值差异的函数来衡量。

在分类中,网络正在为数据分配标签,RMSE 不是计算预测标签质量的合适度量,因为预测不是实数。交叉熵误差是分类问题更合适的度量。

在这种情况下,您可以像这样使用交叉熵误差:

eval_metrics = ['ce']
eval_metrics.append(mx.metric.create('ce'))