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

相关推荐
凌览19 分钟前
有了 25k Star 的MediaCrawler爬虫库加持,三分钟搞定某红书、某音等平台爬取!
前端·后端·python
G.E.N.20 分钟前
开源!RAG竞技场(2):标准RAG算法
大数据·人工智能·深度学习·神经网络·算法·llm·rag
西西弗Sisyphus29 分钟前
如果让计算机理解人类语言- Word2Vec(Word to Vector,2013)
人工智能·word·word2vec
这里有鱼汤31 分钟前
给你的DeepSeek装上实时行情,让他帮你炒股
后端·python·mcp
前端双越老师1 小时前
30 行代码 langChain.js 开发你的第一个 Agent
人工智能·node.js·agent
Tomorrow'sThinker1 小时前
[特殊字符] Excel 读取收件人 + Outlook 批量发送带附件邮件 —— Python 自动化实战
python·excel·outlook
东坡肘子1 小时前
高温与奇怪的天象 | 肘子的 Swift 周报 #092
人工智能·swiftui·swift
JosieBook1 小时前
【Java编程动手学】Java常用工具类
java·python·mysql
Green1Leaves1 小时前
pytorch学习-11卷积神经网络(高级篇)
pytorch·学习·cnn
KaneLogger1 小时前
视频转文字,别再反复拖进度条了
前端·javascript·人工智能