Всем привет, пытаюсь дообучить BertForSequenceClassification из transformers на регрессию (num_labels=1), но outputs у модели - это loss и logits. Не могу разобраться как сделать предсказание, если возвращается результат до софтмакса (как это работает с регрессией?). В статьях везде примеры на классификацию, и это просто np.argsmax(..), а что делать с регрессией?