当前位置: 首页 > news >正文

nn.BCELoss与nn.CrossEntropyLoss

        BCELoss与CrossEntropyLoss都是用于分类问题。可以知道,BCELoss是Binary CrossEntropyLoss的缩写,BCELoss是CrossEntropyLoss的一个特例,只用于二分类问题,而CrossEntropyLoss可以用于二分类,也可以用于多分类,在计算nn.CrossEntropyLoss时,真实的label会被处理成one-hot编码形式

        不过我重新查阅了一下资料,发现同样是处理二分类问题,BCELoss与CrossEntropyLoss是不同的。下面我详细讲一下哪里不同。

1.使用nn.BCELoss需要在该层前面加上Sigmoid函数,公式如下:

l o s s ( X i , y i ) = − w i [ y i l o g x i + ( 1 − y i ) l o g ( 1 − x i ) ] loss(X_{i},y_{i}) = -w_{i}[y_{i}logx_{i} + (1-y_{i})log(1-x_{i})] loss(Xi​,yi​)=−wi​[yi​logxi​+(1−yi​)log(1−xi​)]

 2.使用nn.CrossEntropyLoss会自动加上Sofrmax层,公式如下:

在这里插入图片描述

用法总结:
sigmoid+BCE loss
softmax+CrossEntropyLoss 

http://www.lryc.cn/news/99267.html

相关文章:

  • CSDN浏览如何解决
  • web前端开发小知识
  • Java泛型的简单认识
  • 视频转化为图片或灰度视频
  • 【动态规划刷题 2】使⽤最⼩花费爬楼梯 解码⽅法
  • Python的基本语法
  • Kubernetes那点事儿——存储之存储卷
  • Go语言中‘String’包中的‘Cut‘函数的实现
  • 【JAVASE】顺序和选择结构
  • Oracle恢复删除的数据
  • (无人机方向)ros小白之键盘控制无人机(终端方式)
  • 【python学习笔记】argparse --- 命令行选项、参数和子命令解析器
  • 【Java框架】RPC远程调用
  • 云原生全栈体系(一)
  • 【【51单片机直流电机调速】】
  • 【Spring Boot】
  • 使用docker 部署自己的chatgpt
  • Python适配器模式介绍、使用方法
  • 【数据结构】复杂度
  • 【读点论文】PP-YOLOE: An evolved version of YOLO,面向友好部署的模型设计,为项目后续产业落地提供了更加有效的参考
  • 微服务入门---SpringCloud(二)
  • 51单片机IO口控制
  • ERROR 1064 - You have an error in your SQL syntax;
  • leetcode做题笔记46
  • 快问快答JS面向对象面试题
  • googlenet论文理解
  • OnnxRuntime TensorRT OpenCV::DNN性能对比(YoloV8)实测
  • 【QT 网络云盘客户端】——获取用户文件列表信息
  • 从0到1,无代码开发如何简化产品创新流程
  • select、epoll 的快速核心理解