Hello ,

I’m doing classification using recurrent network and I remember using a metric in SKlearn with random forest that gives the classifier probability of each class , example I have 3 classes this metric gives me the probability of what the classifier thinks this object belongs to which class , Is there a built in function in Pytorch that does this ?

Thank you

If your model returns logits, you could call `F.softmax`

on them, which will yield the class probabilities.

Example:

```
import torch.nn.functional as F
model = nn.Linear(10, 5)
x = torch.randn(1, 10)
output = model(x)
prob = F.softmax(output, dim=1)
```

1 Like