Pytorch 如何得到预测概率？
此代码从模型中获取Pytorch 如何得到预测概率？,pytorch,Pytorch,此代码从模型中获取1或0值 如果我想得到预测的概率 我应该换哪一行 from torch.autograd import Variable results = [] #names = [] with torch.no_grad(): model.eval() print('===============================================start') for num, data in enumerate(test_loader):
10如果我想得到预测的概率
我应该换哪一行
from torch.autograd import Variable
results = []
#names = []
with torch.no_grad():
model.eval()
print('===============================================start')
for num, data in enumerate(test_loader):
#print(num)
print("=====================================================")
imgs, label = data
imgs,labels = imgs.to(device), label.to(device)
test = Variable(imgs)
output = model(test)
#print(output)
ps = torch.exp(output)
print(ps)
top_p, top_class = ps.topk(1, dim = 1)
results += top_class.cpu().numpy().tolist()
model = models.resnet50(pretrained=True)
model.fc = nn.Linear(2048, num_classes)
model.cuda()
模型通常输出原始预测逻辑。若要将它们转换为概率，应使用函数
import torch.nn.功能与nnf相同
# ...
prob=nnf.softmax（输出，尺寸=1）
顶部，顶部等级=概率顶部（1，尺寸=1）
新变量
top\u ptop\u p，top\u class=ps.topk（1，dim=1）top\u p模型torch.exp输出