Edit model card

FinBERT for Financial News Sentiment Regression

DISCLAIMER: This model has been successfully tested with a test set of the same distribution. However, it is not a production-ready model as it probably needs to be updated continuously. Furthermore, the model should have been trained with more than two years of historical data. Additionally, it would need a supplementary assessment on bias, security and consistency.

Introduction

Analyzing the sentiment of financial news is a complex task that requires a large understanding of the financial slang, as well as the knowledge of the context of each one of the companies, and the interactions of the whole economy as an ecosystem.

The FinBERT model binary classifies the sentiment being positive or negative. However, the idea of binary classification is too simple and does not comply with the reality.

RavenPack has an excellent hand-labelled large dataset with a continuous sentiment label variable that goes from -1 to 1. We have collected data from two previous years and tested it with data from the next two weeks. Additionally we have cut the dataset taking only both one year and six months subsamples to see how the model scales with more data, and to know whether more data helps the model or not.

In this repository you can find the different models by changing the branch name. The main branch is the one with the model trained on the whole dataset. We also uploaded the best regressor FinEAS to the Hub: https://huggingface.co/LHF/FinEAS

Note that the predictions of this HF model will go from 0 to 1 being 0.5 neutral, 1 positive and 0 negative.

Evaluation

Dates FinEAS FinBERT
6 months 0.0044 0.0050
12 months 0.0036 0.0034
24 months 0.0033 0.0040
*Evaluated with the next two weeks.

Code

You can find the code for this model in the following link: https://github.com/lhf-labs/finance-news-analysis-bert

Citation

@misc{gutierrezfandino2021fineas,
      title={FinEAS: Financial Embedding Analysis of Sentiment}, 
      author={Asier Gutiérrez-Fandiño and Miquel Noguer i Alonso and Petter Kolm and Jordi Armengol-Estapé},
      year={2021},
      eprint={2111.00526},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}
Downloads last month
13
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.