分类损失函数 (一) torch.nn.CrossEntropyLoss()

1、交叉熵

  • 是一种用于衡量两个概率分布之间的距离或相似性的度量方法。
  • 机器学习中,交叉熵常用于损失函数,用于评估模型的预测结果和实际标签的差异。
  • 公式:

y:真是标签的概率分布,y':模型预测的概率分布

  • 交叉熵损失函数的目的:是最小化预测概率分布和真是概率分布之间的差异,使模型能够更好的学习到数据的内在规律和特征。

2、pytorch中的CrossEntropyLoss()

  • pytorch中的交叉熵包含两部分,softmax和交叉熵计算
  • softmax将预测值转化为概率值
  • torch.nn.CrossEntropyLoss(logits,target)

其中logits预测值是网络输出:\[0.8, 0.5, 0.2, 0.5,

0.2, 0.9, 0.3, 0.2,

0.4, 0.3, 0.7, 0.1,

0.1, 0.2, 0.4, 0.8\]

其中target标签可以是:列表:torch.tensor(\[1, 0, 0, 0,

0, 1, 0, 0,

0, 1, 0, 0,

0, 0, 0, 1], dtype=torch.float)

索引:torch.tensor(0,1, 1, 3, dtype=torch.long)

相关推荐
记忆张量MemTensor1 分钟前
MemOS Skill 上线|一句话即可接入 MemOS Cloud
大数据·数据库·人工智能·typescript·开源
H0311169852 分钟前
关于AI会议音视频转写与智能生成纪要工具的对比分析
人工智能·音视频
视跃科技5 分钟前
多路视频实时解码推理,是怎么做到“逐帧解码 + AI检测 + 目标框叠加上墙“的?
人工智能·目标跟踪·音视频
嘻嘻的AI日记13 分钟前
多库设置+自主选库检索,重塑智能知识管理体系
人工智能
华万通信king15 分钟前
GEO实战:用 llms.txt 给官网打“AI友好“地基(附配置示例)
人工智能·geo
AI云海19 分钟前
深度学习_全连接网络实现回归任务
人工智能·深度学习·回归
wangchen_022 分钟前
PyTorch
人工智能·pytorch·python
code_pgf24 分钟前
SAMLabeler-MNN 项目使用与编译问题解决
人工智能·深度学习·mnn
LoveAmySun30 分钟前
AI智能体如何落地公交营运真实业务
大数据·人工智能
ʜᴇɴʀʏ32 分钟前
AAAI 2026 | KTV:基于聚类与Token剪枝的关键帧选取
数据挖掘·聚类·剪枝