# Returning logits from Trainer.predict()
**URL:**
**Author:**  [@ehalit](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fehalit)%5C
**Post date:** [August 18, 2021, 5:27am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Freturning-logits-from-trainer-predict%2F9314%2F1 "2021-08-18T05:27:08Z")
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.
---
**Author:**  [@sgugger](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fsgugger)%5C
**Post date:** [August 30, 2021, 6:12pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Freturning-logits-from-trainer-predict%2F9314%2F2 "2021-08-30T18:12:22Z")
The predict method does return the logits, as well as the labels.
---
**Author:**  [@ehalit](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fehalit)%5C
**Post date:** [August 31, 2021, 4:53am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Freturning-logits-from-trainer-predict%2F9314%2F3 "2021-08-31T04:53:28Z")
Thank you for the reply, I realized how stupid of a mistake mine was, but I couldn’t delete the post. It surely does return the logits.
---
**Author:**  [@sgugger](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fsgugger)%5C
**Post date:** [August 31, 2021, 10:53am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Freturning-logits-from-trainer-predict%2F9314%2F4 "2021-08-31T10:53:07Z")
No worries, there are no stupid questions 🙂