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:
