search

yya518 / FinBERT

A Pretrained BERT Model for Financial Communications. https://arxiv.org/abs/2006.08097
Apache License 2.0
560 stars 128 forks source link

Is there sentiment score or probability of positive/neutral/negative? #27

Closed petric3 closed 2 years ago

petric3 commented 2 years ago

Thank you for the great library.

I went through example and there are only labels, i.e. positive, neutral and negative for the text's sentiment. I'm wondering if there's any score or probability deciding what text is positive, neutral or negative?

I was looking in attributes, i.e. dir(outputs) or dir(tourch) but did not find any such value, which would be indeed useful to know.

yya518 commented 2 years ago

Yes, FinBERT produces probabilities for each label.