如何在pytorch中找到分类模型结果的得分概率?

Question

我是pytorch的新手，使用这个我已经训练了一个图像分类模型，当我用图像测试模型时，我只得到标签，如果我想得到该类的预测概率，我怎么能得到呢?

test_image = test_image_tensor.view(1,3,300,300)
model.eval()
out = model(test_image)
ps = torch.exp(out)
topk,topclass = ps.topk(1,dim=1)
class_name = idx_to_class[topclass.cpu().numpy()[0][0]]

我使用上面的代码进行预测，它只给出类名，如果我想要预测的标签分数，我怎么能得到它?

任何关于这方面的帮助或建议将不胜感激

Answer 1

概率是预测的软最大值:

class_prob = torch.softmax(out, dim=1)
# get most probable class and its probability:
class_prob, topclass = torch.max(class_prob, dim=1)
# get class names
class_name = idx_to_class[topclass.cpu().numpy()[0][0]]