原文:pytorch中的nn.CrossEntropyLoss()

nn.CrossEntropyLoss 这个损失函数和我们普通说的交叉熵还是有些区别。 x 是模型生成的结果, class 是数据对应的label loss x,class log frac exp x class sum j exp x j x class log sum j exp x j nn.CrossEntropyLoss 的使用方式参见如下代码 ...

2019-12-11 16:13 0 3235 推荐指数:

查看详情

nn.CrossEntropyLoss

nn.CrossEntropyLoss pytorch交叉熵计算方式为: \[H(p,q) = -\sum p(i)logq(i) \] 其中,p为真实值矩阵,q为预测值矩阵 当P使用one-hot embedding时,只有在分类正确时 nn.CrossEntropyLoss ...

Sun Mar 06 00:28:00 CST 2022 0 916
Pytorch学习笔记12----损失函数nn.CrossEntropyLoss()、nn.NLLLoss()

1.CrossEntropyLoss()损失函数 交叉熵主要是用来判定实际的输出与期望的输出的接近程度,为什么这么说呢,举个例子:在做分类的训练的时候,如果一个样本属于第K类,那么这个类别所对应的的输出节点的输出值应该为1,而其他节点的输出都为0,即[0,0,1,0,….0,0],这个数组也就 ...

Tue Aug 04 02:20:00 CST 2020 0 847
torch.nn.CrossEntropyLoss

class torch.nn.CrossEntropyLoss(weight=None, size_average=True, ignore_index=-100, reduce=True) 我这里没有详细解读这个损失函数的各个参数,仅记录一下在sru涉及到的。 sru中代 ...

Thu Dec 07 19:03:00 CST 2017 0 17609
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM