Soft max is output greated than 1
HI team , i am using AutoModelForSequenceClassification , model is roberta , using it for text classification.there are 3 classesoutput is [-3.7550,-4.4172,7.8079]i need to convert this to probabilitiesshould I apply soft max to this to get the probabilities , if i do that i am getting outputs greater than one[9.51,4.90,0.99]
2023-01-11T11:15:54Z
Try specifying the axis you are applying the softmax function on. Like this:torch.nn.functional.softmax(torch.tensor([[-3.7550, -4.4172, 7.8079]]), dim=1)