Использование предварительно обученной модели без корпуса Huggingface для регрессии. Однако метод предсказания возвращает логит

Question

Я использую TFAutoModelForSequenceClassification для регрессии. Обучение и оценка выглядели правильно. Однако метод предсказания возвратил логиты. Как преобразовать логиты в значения регрессии, которые меня интересуют?

Код, который я использовал для обучения модели