Getting probabilities values as more than 1

i am using distilbert for my model using this link-Text classification

In the compute_metrics function when i print the ‘predictions’ i am seeing values greater than 1 also. Since these are predicted probabilities shouldn’t it be less than 1?

The predictions are not probabilities. You can convert them to probabilities by using something like

probabilities = torch.nn.functional.softmax(predictions, dim=-1)

Thanks Panigrah!