Bruno Dalla Costa

icon location Brazil icon location Computer Science - University of Brasilia

Results 1 issues of Bruno Dalla Costa

Is there a way to get the actual probability of each class from the model prediction?

3 comment

``` from openprompt import PromptForClassification promptModel = PromptForClassification( template = promptTemplate, plm = plm, verbalizer = promptVerbalizer, ) import torch promptModel.eval() with torch.no_grad(): for batch in data_loader: logits = promptModel(batch)...