Pretrained model used for regression returns logits during prediction
I’ve been trying to use the TFAutoModelForSequenceClassification class with the bert-uncased model for a regression task. I was able to fit the model with mae loss. However, when I use the predict method of the model, it returns logits. I am not sure what to do with it as it is not a classification task.Here is my code:image961×709 28 KB