PyTorch笔记--均方误差(MSE)

均方误差(Mean Squared Error)是度量模型性能的一种方法。

假设m是样本集的总个数

 

 是第i个样本的预测值，是第i个样本的真实值。

pytorch中的均方误差函数

torch.nn.functional.mse_loss(
input,  # 预测
target, # 目标
)

 

代码实现：

>>>import torch
>>>import torch.nn.functional as F
>>> x = torch.randn(5, 10)
>>> w = torch.randn(10, 10)
>>> logits = x @ w.t()
>>> pred = torch.sigmoid(logits) # 预测值
>>> loss = F.mse_loss(pred, target) # 计算mse_loss