分类损失函数 (一) 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)

相关推荐
把所有砖敲烂15 小时前
DeepSeek V4正式版发布,教你在Codex中配置Flash
人工智能·产品
wyg_03111316 小时前
从0搭建极简transformer大模型
人工智能·深度学习·transformer
老王以为16 小时前
走进 AI Agent
前端·人工智能·架构
dong_junshuai16 小时前
每天一个开源项目#59 AirLLM:27.6K星,3.72GB显存跑2.8T模型
人工智能·程序员·github
文心快码BaiduComate16 小时前
Comate 已支持DeepSeek-V4-Flash 正式版
人工智能·程序员
MartinYeung516 小时前
[论文学习]OSReward:为跨平台计算机使用奖励模型建立标准化评估
人工智能·学习
数字供应链安全产品选型16 小时前
悬镜安全打造“中国版 Anthropic Mythos”:以AI治理AI,重构新一代数字供应链安全
人工智能·安全·重构
Tangyuewei17 小时前
2026 下半年:从更快到更可控
人工智能
世人万千丶17 小时前
鸿蒙Crash高级捕获与异常监控:全局异常兜底/崩溃栈解析/符号表还原/智能聚类/闭环修复
学习·机器学习·华为·数据挖掘·harmonyos·鸿蒙·聚类
qq_3164110317 小时前
AI 情感陪伴智能潮玩软硬件一体化开发案例
人工智能·python