张量流中的tf.estimator.DNNRegressor的平均损失是否与MSE相同?

时间:2018-10-08 06:17:10

标签: python tensorflow

model.evaluate of tensorflow中,有一个名为“ average_loss ”的变量。标签和预测之间的MSE是否相同?但是,在tf.losses中,还有另一个功能mean_squared_error。哪个是正确的MSE?

1 个答案:

答案 0 :(得分:0)

根据模型(see here)计算损失。

您的模型很可能对损失函数使用均方误差,因此 mean_squared_error 也是损失变量。

相反,平均损失将是最后N次迭代的平均损失,因为有时训练是在小批量的情况下进行的,并且损失可能很吵。