Questions about regression

ProsusAI / finBERT

Financial Sentiment Analysis with BERT

Apache License 2.0

1.42k stars 413 forks source link

Questions about regression #63

Closed Jacobluke- closed 2 years ago

Jacobluke- commented 2 years ago

If I want to use FinBERT for regression, what should I do with the prepare_model() function? Since there is no classification label like ['positive', 'neutral', 'negative'] to input.

doguaraci commented 2 years ago

Hi, you don't need to change anything. The code already supports regression. You need to set the output mode as regression when you instantiate the config.