![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
损失函数
BlackMan_阿伟
不积跬步无以至千里
展开
-
Pyroch损失函数之BCELoss
这也是最近在做的一个模型中的损失函数,所有就一探究竟了解一下这个函数。首先它是一个二分类损失函数。可以是单标签的损失函数也可是多标签的损失函数。1、单标签这个图像是不是猫:1代表是,0代表不是。这就是单标签的二分类问题。2、多标签和单标签一样,不过这个是多标签而已,所以它的label就是[1,1]。第一列代表有没有狗,第二列代表有没有猫。所以也称之为二进制交叉熵。3、BCELoos损失函数class torch.nn.BCELoss(weight: Optional[to.原创 2020-07-02 15:46:28 · 19149 阅读 · 8 评论 -
Pytorch中的损失函数:CrossEntropyLoss和NLLLoss的区别
在网上也浏览了很多的博客,很多博主都从熵和信息量开始介绍,因为最近在使用损失函数过程中遇到一点问题,因此就想一探究竟到底pytorch中的交叉熵损失函数是如何计算的,所有最简单的从公式和代码开始看。接下来直接介绍这两个函数:1、NLLLossclass torch.nn.NLLLoss(weight=None, size_average=True)作用:训练一个n类的分类器 参数 weight:可选的,应该是一个tensor,里面的值对应类别的权重,如果样本不均衡的话,这个参数非常有用,长原创 2020-06-29 16:32:55 · 2736 阅读 · 2 评论