机器学习损失函数总结
作者:互联网
文章目录
前言
机器学习损失函数总结
一、二值交叉熵损失函数
- 主要用来对目标分类为0-1的情况的分类任务进行损失计算的
- 输入的是一个预测值和对应的一个目标值,输出损失结果。如果输入的是一系列的预测值,那么就输入对应的一系列目标值,然后求每个
预测-目标
对的损失,再对最后的结果求平均。
- 输入的是一个预测值和对应的一个目标值,输出损失结果。如果输入的是一系列的预测值,那么就输入对应的一系列目标值,然后求每个
- pytorch的损失函数有
# nn的
torch.nn.BCELoss(weight: Optional[torch.Tensor
标签:总结,机器,函数,nn,torch,损失,二值,输入 来源: https://blog.csdn.net/qq_37771209/article/details/110733831