Views
No views yet
| Metric / Dataset | FinBERT (Multi-Task) | FinBERT (LoRA) |
|---|---|---|
| Overall Accuracy | 85.4% | 83.2% |
| Macro F1-Score | 0.83 | 0.80 |
| Financial PhraseBank (News) | 95.9% | 97.1% |
| Twitter Financial News | 83.3% | 80.5% |
| FiQA (Forums) | 81.5% | 72.6% |
Note: For edge deployment or low-memory environments, check out the LoRA version which reduces storage by 99% (5MB vs 420MB).
bert-base backbone with two task-specific heads:Negative/Neutral/Positive (Optimized for News & Twitter).pipeline or AutoModel:1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load the model and tokenizer
5model_name = "pmatorras/financial-sentiment-multi-task"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9# Inference
10text = "The stock market rally is driven by strong tech earnings."
11inputs = tokenizer(text, return_tensors="pt")
12
13with torch.no_grad():
14 outputs = model(**inputs)
15 probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
16
17print(probabilities)