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?
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!