Getting probabilities values as more than 1
i am using distilbert for my model using this link-Text classificationIn 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?
2023-11-24T12:33:42Z
The predictions are not probabilities. You can convert them to probabilities by using something likeprobabilities = torch.nn.functional.softmax(predictions, dim=-1)
2023-11-24T22:16:21Z
Thanks Panigrah!