Hello! I would like to perform some operations on the output probability distributions of an AutoModelForSequenceClassification model so, I was wondering if it is possible to return the logits rather than predicted class labels from the transformers.Trainer.predict() method.

Returning logits from Trainer.predict()

sgugger August 31, 2021, 10:53am 4

No worries, there are no stupid questions

1 Like

Topic		Replies	Views
Confusion about trainer.predict(dataset['test']) output 🤗Transformers	0	532	November 3, 2022
Is there a way to get the logits as output in zero shot classification Pipeline? 🤗Transformers	0	480	July 30, 2021
How to use the multiple output of the model while calling Trainer 🤗Transformers	0	489	August 10, 2021
Trainer predict or evulate does not return softmax or sigmoid value 🤗Transformers	1	992	May 11, 2022
How to get the logits value from SequenceClassifierOutput? Beginners	0	609	May 26, 2022