pytorch 均方误差损失函数

均方误差损失函数主要用于回归问题。它计算预测值与真实值之间差的平方,然后取平均值。这个损失函数通过惩罚大的误差,使得模型在训练时更加注重减少较大的偏差。

复制代码
import torch
import torch.nn as nn

# 创建预测值和实际值张量
predicted = torch.tensor([0.5, 0.3, 0.2], requires_grad=True)
actual = torch.tensor([0.6, 0.1, 0.2])

# 定义 MSE 损失函数
criterion = nn.MSELoss()

# 计算损失
loss = criterion(predicted, actual)
print(loss.item())  # 输出损失值

解释:

  • predicted 是模型的预测输出,actual 是对应的真实值。
  • nn.MSELoss() 定义了均方误差损失函数。
  • loss = criterion(predicted, actual) 计算预测值和实际值之间的均方误差。
  • .item() 用于从单个元素张量中提取数值。
参考

MSELoss --- PyTorch 2.4 documentation

相关推荐
拓端研究室1 小时前
视频讲解|核密度估计朴素贝叶斯:业务数据分类—从理论到实践
人工智能·分类·数据挖掘
灵智工坊LingzhiAI1 小时前
人体坐姿检测系统项目教程(YOLO11+PyTorch+可视化)
人工智能·pytorch·python
昨日之日20061 小时前
Video Background Remover V3版 - AI视频一键抠像/视频换背景 支持50系显卡 一键整合包下载
人工智能·音视频
SHIPKING3932 小时前
【机器学习&深度学习】什么是下游任务模型?
人工智能·深度学习·机器学习
子燕若水6 小时前
Unreal Engine 5中的AI知识
人工智能
极限实验室7 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿7 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫7 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手7 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
AI生存日记8 小时前
百度文心大模型 4.5 系列全面开源 英特尔同步支持端侧部署
人工智能·百度·开源·open ai大模型