Downloads · 30 days
14
0% of all-time downloads
stfamod/fine-tuned-bert-financial-sentiment-analysis
fine-tuned-bert-financial-sentiment-analysis is a text classification model from stfamod. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This NLP model is fine-tuned with a focus on analyzing sentiment in financial text and news headlines. It was fine-tuned using the bert-base-uncased model on the financialphrasebank and auditorsentiment datasets.
Downloads · 30 days
14
0% of all-time downloads
All-time downloads
8.1K
Public
Repo size
876 MB
Likes
1
Public
Click a slice to open those files.
.bin438 MB · 100%
From the Hugging Face model README
This NLP model is fine-tuned with a focus on analyzing sentiment in financial text and news headlines. It was fine-tuned using the bert-base-uncased model on the financial_phrasebank and auditor_sentiment datasets.
Accuracies:
financial_phrasebank: 0.993
auditor_senitment: 0.974
Learning Rate: 2e-05
Train Batch Size: 16
Eval Batch Size: 16
Random Seed: 42
Optimizer: AdamW-betas(0.9, 0.999)
Learning Rate Scheduler: Linear
Number of Epochs: 6
Number of Warmup Steps: 0.2 * Number of Training Steps
from transformers import pipeline
pipe = pipeline("sentiment-analysis", model="mstafam/fine-tuned-bert-financial-sentimental-analysis")
text = "Example company has seen a 5% increase in revenue this quarter."
print(pipe(text))
[{'label': 'Positive', 'score': 0.9993795156478882}]