NLP for Economics 1.2
Collection
NLP tools for sentiment analysis and relevance detection
•
4 items
•
Updated
This model is a fine-tuned version of samchain/econo-sentence-v2 on the Financial Phrase Bank dataset from FinanceMTEB. The full model is trained using a small learning rate isntead of freezing the encoder. Hence, you should not use the encoder of this model for a task other than sentiment analysis.
It achieves the following results on the evaluation set:
The base model is a sentence-transformers model built from EconoBert.
This model is trained to provide a useful tool for sentiment analysis in finance.
The dataset is directly downloaded from the huggingface repo of the FinanceMTEB. The preprocessing consisted of tokenizing to a fixed sequence length of 512 tokens.
The following hyperparameters were used during training:
Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
---|---|---|---|---|---|---|---|
0.5992 | 1.0 | 158 | 0.4854 | 0.805 | 0.7692 | 0.8108 | 0.805 |
0.0985 | 2.0 | 316 | 0.1293 | 0.962 | 0.9619 | 0.9619 | 0.962 |
Base model
google-bert/bert-base-uncased