1 二分类交叉熵损失函数
之前已经介绍过交叉熵损失函数【Pytorch基础】torch.nn.CrossEntropyLoss损失函数介绍,现在再来看一下,它的一个变体,二分类交叉熵损失函数。由于是多分类问题,在之前文章中举的例子中,模型预测的样本概率是一个向量。如果是二分类任务的话,因为只有正例和负例,且两者的概率和是1,所以不需要预测一个向量,只需要预测概率。
则损失函数定义如下:
L o s s = − ( y ⋅ log ( y ^ ) + ( 1 − y ) ⋅ log ( 1 − y ^ ) ) Loss=-(y \cdot \log (\hat{y})+(1-y) \cdot \log (1-\hat{y})) Loss=−(y⋅log(y^)+(1−y)⋅log(1

本文介绍了二分类交叉熵损失函数的基本概念及其计算方法,并通过具体示例进行演示。此外,还对比了BCELoss与BCEWithLogitsLoss两种损失函数的不同之处,最后给出了在PyTorch中实现二分类交叉熵损失的代码示例。
最低0.47元/天 解锁文章
864

被折叠的 条评论
为什么被折叠?



