yiyanghkust/finbert-tone is a forked repo from huggingface. License: None
Go to file
Yi 4921590d3c Update README.md 2022-10-17 00:35:39 +00:00
.gitattributes initial commit 2021-07-23 07:44:21 +00:00
README.md Update README.md 2022-10-17 00:35:39 +00:00
config.json captialize label 2022-05-13 15:05:17 +08:00
pytorch_model.bin upload finbert-tone 2021-07-28 21:53:52 +08:00
tf_model.h5 Add TF weights (#2) 2022-06-09 12:05:27 +00:00
vocab.txt upload finbert-tone 2021-07-28 21:53:52 +08:00

README.md

language tags widget
en
financial-sentiment-analysis
sentiment-analysis
text
growth is strong and we have plenty of liquidity

FinBERT is a BERT model pre-trained on financial communication text. The purpose is to enhance financial NLP research and practice. It is trained on the following three financial communication corpus. The total corpora size is 4.9B tokens.

  • Corporate Reports 10-K & 10-Q: 2.5B tokens
  • Earnings Call Transcripts: 1.3B tokens
  • Analyst Reports: 1.1B tokens

More technical details on FinBERT: Click Link

This released finbert-tone model is the FinBERT model fine-tuned on 10,000 manually annotated (positive, negative, neutral) sentences from analyst reports. This model achieves superior performance on financial tone analysis task. If you are simply interested in using FinBERT for financial tone analysis, give it a try.

If you use the model in your academic work, please cite the following paper:

Huang, Allen H., Hui Wang, and Yi Yang. "FinBERT: A Large Language Model for Extracting Information from Financial Text." Contemporary Accounting Research (2022).

How to use

You can use this model with Transformers pipeline for sentiment analysis.

from transformers import BertTokenizer, BertForSequenceClassification
from transformers import pipeline

finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone',num_labels=3)
tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-tone')

nlp = pipeline("sentiment-analysis", model=finbert, tokenizer=tokenizer)

sentences = ["there is a shortage of capital, and we need extra financing",  
             "growth is strong and we have plenty of liquidity", 
             "there are doubts about our finances", 
             "profits are flat"]
results = nlp(sentences)
print(results)  #LABEL_0: neutral; LABEL_1: positive; LABEL_2: negative