首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >零与一之间的损失函数值

零与一之间的损失函数值
EN

Stack Overflow用户
提问于 2022-04-13 18:09:25
回答 1查看 613关注 0票数 0

哪个损失函数只返回0到1之间的值?

问题是关于二进制分类(使用pytorch)的损失。

通过尝试这些功能:

代码语言:javascript
复制
loss = torch.nn.functional.binary_cross_entropy(output, target)

代码语言:javascript
复制
loss_function = torch.nn.BCELoss()
loss = loss_function(output, target)

返回值有时大于1。

EN

回答 1

Stack Overflow用户

发布于 2022-04-14 10:52:54

BCE的损失函数方程为:

对于每一个x_i,y_i,我们都会搞砸:

l_i =−w_iy_i⋅log(x_i)+(1−y_i)⋅log(1−x_i)

BCE损失的默认值是所有l_i的平均值,您可以通过在BCE function.

  • notice的参数中输入reduction='sum'来更改它,即日志函数的输出在此范围内(-infinity,0),这与-w_i相乘;w_i在训练过程中具有一个正值。总之,损失函数的值可以是>1

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71862081

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档