使用softmax求损失的过程中,需要经历以下几步:
1.应用softmax函数; 2.取对数; 3.计算损失
其中第1、2步可以使用torch.nn.LogSoftmax()完成,第3步可以使用torch.nn.NLLLoss完成
也可以使用torch.nn.CrossEntropyLoss一次性完成3步操作,构成一个完整的softmax分类器。(使用最多!!!)
CLASS torch.nn.CrossEntropyLoss(weight=None, size_average=None, ignore_index=-100, reduction=‘mean’)
在使用torch.nn.CrossEntropyLoss之前,最后一层不再需要激活函数(无需非线性变换)
在使用torch.nn.CrossEntropyLoss时,样本的真实类别Y应是一个LongTensor,值为各样本的类别标号。
torch.nn.CrossEntropyLoss使用举例:
import torch
import math
criterion = torch.nn.CrossEntropyLoss()
Y = torch.LongTensor([2,0,1]) # 3个样本是真实标记分别为2,0,1
Y_pred1 = torch.Tensor([[0.1,0.2,0.9],
[1.1,0.1,0.2],
[0.2,2.1,0.1]]) # 注意这里没有经过激活函数(即softmax函数),而是直接扔进CrossEntropyLoss里
Y_pred2 = torch.Tensor([[0.8,0.2,0.3],
[0.2,0.3,0.5],
[0.2,0.2,0.5]])
loss1 = criterion(Y_pred1,Y) # 注意这里要把Y_pred1放前面,否则会报错
loss2 = criterion(Y_pred2,Y) # 注意这里要把Y_pred2放前面,否则会报错
print("loss1 = ", loss1, "loss2 = ", loss2) # loss1 = tensor(0.4966) loss2 = tensor(1.2389)
a = math.exp(0.9) / (math.exp(0.1)+math.exp(0.2)+math.exp(0.9))
a = -math.log(a) # 样本1的损失
b = math.exp(1.1) / (math.exp(1.1)+math.exp(0.1)+math.exp(0.2))
b = -math.log(b) # 样本2的损失
c = math.exp(2.1) / (math.exp(0.2)+math.exp(2.1)+math.exp(0.1))
c = -math.log(c) # 样本3的损失
print((a+b+c)/3) # 0.49663528926362077,和loss1一致
版权声明:本文为zyk9916原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。