Loadingβ¦
Loadingβ¦
ProsusAI/finbertA multi-signal macroeconomic sentiment pipeline that combines fine-tuned transformer ensembles with financial dictionaries and topic routing to produce structured sentiment analysis for financial news, central bank communications, climate/ESG reports, and social media.
This repo hosts the FinBERT head (109M params, fine-tuned from ProsusAI/finbert) β the default routing target and backbone of the full pipeline. The complete system includes three additional transformer heads, four dictionary scorers, and a keyword-based topic router.
ββββββββββββββββββββββββββββββββββββ
β Input Text β
ββββββββββββββββ¬ββββββββββββββββββββ
β
ββββββββββββββββββββββΌβββββββββββββββββββββ
βΌ βΌ βΌ
βββββββββββββββββββ βββββββββββββββββββ βββββββββββββββββ
β Topic Router β β Dictionary β β Language β
β (keywords) β β Scorers (Γ4) β β Detection β
ββββββββββ¬βββββββββ ββββββββββ¬βββββββββ βββββββββ¬ββββββββ
β β β
ββββββββββΌβββββββββββββββββββββΌβββββββββββββββββββββΌβββββββββ
β Head Selection β
β policy β RoBERTa-Large β climate β ClimateBERT β
β financial β FinBERT β
β non-English β XLM-RoBERTa β
ββββββββββ¬βββββββββββββββββββββββββββββββββββββββββ¬βββββββββ
β β
ββββββββββΌβββββββββ ββββββββββΌβββββββββ
β Transformer β β Dictionary β
β Score [-1,+1] β β Composite β
ββββββββββ¬βββββββββ ββββββββββ¬βββββββββ
β β
ββββββββββββββββ¬ββββββββββββββββββββββββββ
βΌ
ββββββββββββββββββββββββββββββββ
β Weighted Fusion β
β (crisis-adaptive weights) β
ββββββββββββββββ¬ββββββββββββββββ
βΌ
ββββββββββββββββββββββββββββββββ
β MacroSentimentResult β
β β’ macro_sentiment [-1,+1] β
β β’ policy_stance [-1,+1] β
β β’ crisis_signal [0,1] β
β β’ climate_sentiment [-1,+1] β
β β’ confidence [0,1] β
β β’ detected_domain β
ββββββββββββββββββββββββββββββββ
The fusion weights are crisis-adaptive: when the crisis dictionary fires strongly, more weight shifts to the dictionary composite (up to 75% dict / 25% transformer), since crisis language often carries clearer signal through keywords than neural softmax probabilities.
| Head | Model | Params | Base | Role |
|---|---|---|---|---|
| FinBERT β | peyterho/finbert-macro-sentiment | 109M | ProsusAI/finbert | Default β financial news, tweets |
| RoBERTa-Large | peyterho/financial-roberta-large-macro-sentiment | 355M | soleimanian/financial-roberta-large-sentiment | Policy/macro text |
| ClimateBERT | peyterho/climatebert-macro-sentiment | 82M | climatebert/distilroberta-base-climate-sentiment | Climate/ESG text |
| XLM-RoBERTa | cardiffnlp/twitter-xlm-roberta-base-sentiment-multilingual | 278M | β | Non-English text (8+ languages) |
| Dictionary | Based On | Signal |
|---|---|---|
| Loughran-McDonald | Loughran & McDonald (2011) | Financial polarity, subjectivity |
| Henry | Henry (2008) β earnings tone | Earnings press release tone |
| Climate Exposure | Sautner et al. (2023) style | Climate risk vs. opportunity density |
| Macro Policy | Custom | Hawkish/dovish stance, crisis intensity, uncertainty |
There are two ways to use this model β pick the one that fits your needs:
If you just need positive/negative/neutral labels, use the model directly β no repo cloning required:
# pip install transformers torch
from transformers import pipeline
classifier = pipeline("text-classification", model="peyterho/macro-sentiment-finbert", top_k=None)
result = classifier("The Federal Reserve signaled a pause in rate hikes amid cooling inflation.")
print(result)
# [[{'label': 'positive', 'score': 0.72}, {'label': 'neutral', 'score': 0.21}, {'label': 'negative', 'score': 0.07}]]
For the complete system β topic routing, policy stance, crisis signals, climate scoring, and dictionaries β you need to clone this repo since the pipeline code lives inside it:
git clone https://huggingface.co/peyterho/macro-sentiment-finbert
cd macro-sentiment-finbert
pip install -r requirements.txt
from macro_sentiment import MacroSentimentPipeline
pipe = MacroSentimentPipeline(device="cpu")
# Financial news β auto-routes to FinBERT
result = pipe("Markets rallied on strong earnings, with the S&P 500 hitting record highs.")
print(result.summary())
# Sentiment: Positive (+0.612) | Policy: Neutral (+0.000) | Crisis: Normal (0.000) | Domain: financial_news
# Central bank communication β auto-routes to RoBERTa-Large
result = pipe("The ECB raised rates by 25 basis points, citing persistent inflation pressures.")
print(result.summary())
# Sentiment: Negative (-0.348) | Policy: Hawkish (+0.714) | Crisis: Normal (0.000) | Domain: policy
# Climate/ESG text β auto-routes to ClimateBERT
result = pipe("The company committed to net-zero emissions by 2040 through renewable energy investments.")
print(result.summary())
# Sentiment: Positive (+0.445) | Policy: Neutral (+0.000) | Crisis: Normal (0.000) | Domain: climate | Climate: Opportunity (exp=0.60)
# Non-English text β auto-routes to XLM-RoBERTa
result = pipe("Die EZB signalisiert Geduld bei kΓΌnftigen Zinssenkungen.")
print(result.summary())
Copy-paste these cells into a notebook. Both work on Colab (free tier) and Kaggle.
!pip install -q transformers torch
from transformers import pipeline
classifier = pipeline("text-classification", model="peyterho/macro-sentiment-finbert", top_k=None)
# Try it
texts = [
"Tesla shares surged 15% after crushing earnings expectations.",
"The Federal Reserve raised rates by 75bps citing persistent inflation.",
"Markets crashed amid recession fears and massive layoffs.",
"The company reported quarterly results in line with analyst estimates.",
]
for text in texts:
result = classifier(text)[0]
top = max(result, key=lambda x: x["score"])
print(f'{top["label"]:>8s} ({top["score"]:.2f}) {text}')
# Install dependencies and clone the repo
!pip install -q transformers torch pysentiment2 scikit-learn numpy pandas datasets accelerate huggingface_hub
!git clone https://huggingface.co/peyterho/macro-sentiment-finbert /content/macro-sentiment-finbert
import sys
sys.path.insert(0, "/content/macro-sentiment-finbert")
from macro_sentiment import MacroSentimentPipeline
pipe = MacroSentimentPipeline(device="cpu")
texts = [
"Markets rallied on strong earnings, with the S&P 500 hitting record highs.",
"The ECB raised rates by 25 basis points, citing persistent inflation pressures.",
"The company committed to net-zero emissions by 2040 through renewable energy investments.",
"Credit markets froze as contagion fears spread across European banks.",
"Die EZB signalisiert Geduld bei kΓΌnftigen Zinssenkungen.",
]
for text in texts:
result = pipe(text)
print(result.summary())
print(f" β {text}\n")
result = pipe("Fed signals two more rate cuts before year-end, a dovish surprise that lifted equities.")
print(f"Macro sentiment: {result.macro_sentiment:+.3f}")
print(f"Financial sentiment:{result.financial_sentiment:+.3f}")
print(f"Policy stance: {result.policy_stance:+.3f} (negative=dovish, positive=hawkish)")
print(f"Crisis signal: {result.crisis_signal:.3f}")
print(f"Climate sentiment: {result.climate_sentiment:+.3f}")
print(f"Uncertainty: {result.uncertainty:.3f}")
print(f"Confidence: {result.confidence:.3f}")
print(f"Domain: {result.detected_domain}")
print(f"Head used: {result.head_used}")
print(f"LM polarity: {result.lm_polarity:+.3f}")
print(f"Henry polarity: {result.henry_polarity:+.3f}")
print(f"Climate exposure: {result.climate_exposure:.3f}")
import pandas as pd
headlines = [
"Strong jobs report pushes markets to record highs",
"Tech earnings mixed as AI spending soars",
"Fed signals patience on rate cuts, markets dip",
"Retail sales disappoint, recession fears resurface",
"Green bond issuance hit $500B as investors pivot to sustainable fixed income",
"Bank of Japan holds rates steady in surprise decision",
]
results = pipe.score_batch(headlines, mode="routed")
df = pd.DataFrame([{
"text": t,
"sentiment": r.macro_sentiment,
"policy": r.policy_stance,
"crisis": r.crisis_signal,
"climate": r.climate_sentiment,
"domain": r.detected_domain,
"head": r.head_used,
} for t, r in zip(headlines, results)])
print(df.to_string(index=False))
# No transformer models loaded β uses only the four dictionaries
result = pipe.score("The central bank cut rates amid fears of a deepening recession.", mode="dict_only")
print(result.summary())
print(f"LM polarity: {result.lm_polarity:+.3f}")
print(f"Henry polarity: {result.henry_polarity:+.3f}")
print(f"Policy stance: {result.policy_stance:+.3f}")
print(f"Crisis signal: {result.crisis_signal:.3f}")
π‘ Notes:
- The full pipeline lazy-loads transformer models on first use. First call takes 30β60 seconds to download (~800MB across 4 models). Subsequent calls are fast.
mode="routed"(default) loads only 1 model per call.mode="all"loads all 4 models (~2GB RAM).- For Colab free tier,
mode="routed"works fine. Formode="all", use a GPU runtime to avoid OOM.- On Kaggle, enable "Internet" in notebook settings (Settings β Internet β On) so models can download.
Every call returns a MacroSentimentResult with these fields:
| Field | Range | Description |
|---|---|---|
macro_sentiment |
[-1, +1] | Overall macroeconomic sentiment (weighted fusion of transformer + dictionary) |
financial_sentiment |
[-1, +1] | Financial-domain sentiment from the selected transformer head |
policy_stance |
[-1, +1] | Monetary policy orientation: -1 = very dovish, +1 = very hawkish |
climate_sentiment |
[-1, +1] | Climate outlook: -1 = risk, +1 = opportunity |
crisis_signal |
[0, 1] | Crisis language intensity (recession, contagion, bank failure, etc.) |
uncertainty |
[0, 1] | Uncertainty/volatility language density |
confidence |
[0, 1] | Pipeline confidence (based on head agreement, topic match, uncertainty) |
detected_domain |
str | Routed domain: financial_news, policy, climate, social, ensemble |
head_used |
str | Which transformer head was selected |
lm_polarity |
[-1, +1] | Loughran-McDonald polarity score |
henry_polarity |
[-1, +1] | Henry earnings tone score |
climate_exposure |
[0, 1] | Climate keyword density (Sautner-style) |
# "routed" (default) β topic router selects best head
result = pipe.score("text", mode="routed")
# "all" β runs ALL heads and averages composite scores
result = pipe.score("text", mode="all")
# "dict_only" β dictionary signals only, no transformer inference
result = pipe.score("text", mode="dict_only")
All three transformer heads were fine-tuned on the same combined dataset of 5 public financial/climate sentiment corpora:
| Dataset | Domain | Samples | Label Mapping |
|---|---|---|---|
| nickmuchi/financial-classification | Financial PhraseBank | ~4,800 train / ~1,200 test | negative / neutral / positive |
| zeroshot/twitter-financial-news-sentiment | Financial tweets | ~9,900 train / ~2,500 val | bearish β neg, bullish β pos, neutral |
| FinanceInc/auditor_sentiment | Auditor reports | ~3,600 train / ~900 test | negative / neutral / positive |
| pauri32/fiqa-2018 | Financial QA + microblog | ~938 train+val / ~235 test | Continuous score thresholded at Β±0.15 |
| climatebert/climate_sentiment | Climate reports | ~1,000 train / ~500 test | risk β neg, neutral, opportunity β pos |
All datasets were unified to a consistent 3-class schema: 0=negative, 1=neutral, 2=positive.
| Hyperparameter | Value |
|---|---|
| Base model | ProsusAI/finbert |
| Learning rate | 2e-5 |
| Batch size | 16 Γ 4 gradient accumulation = 64 effective |
| Epochs | 2 |
| Scheduler | Cosine with 31 warmup steps |
| Optimizer | AdamW (fused) |
| Max length | 512 |
| Seed | 42 |
| Epoch | Train Loss | Val Loss | Accuracy | F1 (macro) | F1 (weighted) |
|---|---|---|---|---|---|
| 1 | 1.6807 | 0.3761 | 0.8246 | 0.8058 | 0.8300 |
| 2 | 1.1679 | 0.3450 | 0.8486 | 0.8325 | 0.8515 |
Note: The higher training loss reflects the FinBERT label ordering (positive=0, negative=1, neutral=2) which differs from the unified schema β a label remapping is applied at training time.
| Model | Accuracy | F1 (macro) | F1 (weighted) |
|---|---|---|---|
| RoBERTa-Large (355M) | 0.9130 | 0.9023 | 0.9137 |
| FinBERT (109M) β | 0.8973 | 0.8813 | 0.8984 |
| ClimateBERT (82M) | 0.8885 | 0.8716 | 0.8898 |
| Dict-only baseline (GBT) | 0.6693 | 0.5781 | 0.6500 |
| Dict-only baseline (rules) | 0.5684 | 0.5277 | 0.5784 |
Evaluated on Jean-Baptiste/financial_news_sentiment_mixte_with_phrasebank_75 β not in the training mix.
| Model | Accuracy | F1 (macro) |
|---|---|---|
| RoBERTa-Large | 0.9414 | 0.9357 |
| ClimateBERT | 0.9248 | 0.9213 |
| FinBERT β | 0.9236 | 0.9134 |
Evaluated on ic-fspml/stock_news_sentiment β 5-class mapped to 3-class. Not in the training mix.
| Model | Accuracy | F1 (macro) |
|---|---|---|
| RoBERTa-Large | 0.7211 | 0.7265 |
| FinBERT β | 0.6781 | 0.6765 |
| ClimateBERT | 0.6472 | 0.6441 |
Performance drops on stock news headlines are expected β these are short, noisy texts with 5β3 class mapping, representing a significant domain shift from the training data.
This section walks you through fine-tuning this model (or any of the ensemble heads) on your own labelled data. The repo includes a ready-to-use finetune.py script that handles label remapping, class-weighted loss, and evaluation automatically.
git clone https://huggingface.co/peyterho/macro-sentiment-finbert
cd macro-sentiment-finbert
pip install -r requirements.txt
pip install evaluate # needed for metrics during training
Create a file with two columns: one for text, one for labels. Supported formats: CSV, TSV, JSON, or JSONL.
Labels can be strings or integers. The script automatically maps them to the unified 3-class schema (0=negative, 1=neutral, 2=positive):
| Accepted strings | Maps to |
|---|---|
"negative", "neg", "bearish", "risk" |
0 (negative) |
"neutral", "neut", "mixed" |
1 (neutral) |
"positive", "pos", "bullish", "opportunity" |
2 (positive) |
Or just use integers: 0, 1, 2.
my_labels.csv)headline,sentiment
"Markets rallied on strong quarterly earnings",positive
"Fed raises rates citing persistent inflation",negative
"Company reports results in line with expectations",neutral
"Oil prices surge amid Middle East tensions",negative
"Record-breaking IPO signals strong investor confidence",positive
my_labels.jsonl){"text": "Markets rallied on strong quarterly earnings", "label": "positive"}
{"text": "Fed raises rates citing persistent inflation", "label": "negative"}
{"text": "Company reports results in line with expectations", "label": "neutral"}
π‘ How much data? Even 200β500 labelled examples can meaningfully improve performance on your specific domain. 1,000+ is ideal. The script automatically applies class-weighted loss to handle imbalanced label distributions β so don't worry if you have more of one class than another.
python -m macro_sentiment.finetune \
--data my_labels.csv \
--text-column headline \
--label-column sentiment
This saves the fine-tuned model to ./custom-finetuned/.
python -m macro_sentiment.finetune \
--data my_labels.csv \
--text-column headline \
--label-column sentiment \
--base-model peyterho/macro-sentiment-finbert \
--output my-username/my-custom-finbert \
--push-to-hub \
--epochs 4 \
--lr 2e-5 \
--batch-size 32 \
--max-length 128
The script will:
| Argument | Default | Description |
|---|---|---|
--data |
required | Path to your data file (.csv, .tsv, .json, .jsonl) |
--text-column |
required | Column name containing the text |
--label-column |
required | Column name containing the labels |
--base-model |
peyterho/finbert-macro-sentiment |
Base model to fine-tune (see table below) |
--output |
./custom-finetuned |
Output directory or Hub model ID (e.g. my-org/my-model) |
--push-to-hub |
false |
Push the final model to Hugging Face Hub |
--epochs |
4 |
Number of training epochs |
--lr |
2e-5 |
Learning rate |
--batch-size |
32 |
Per-device batch size (effective batch = batch-size Γ gradient_accumulation) |
--max-length |
128 |
Max token sequence length (increase to 256 or 512 for long texts) |
--val-split |
0.15 |
Fraction of data for validation (stratified by label) |
--no-class-weights |
false |
Disable automatic class-weighted loss |
--seed |
42 |
Random seed for reproducibility |
You can fine-tune any of the ensemble heads. Pick the one closest to your domain:
| Use case | Recommended --base-model |
Why |
|---|---|---|
| Financial news, tweets, earnings | peyterho/macro-sentiment-finbert (default) |
Pre-trained on financial corpora, 109M params, fast |
| Central bank / policy / macro reports | peyterho/financial-roberta-large-macro-sentiment |
Largest head (355M), best on policy language |
| Climate / ESG / sustainability reports | peyterho/climatebert-macro-sentiment |
Pre-trained on climate text, smallest (82M) |
| Starting from original base (no macro fine-tune) | ProsusAI/finbert |
Use if you want to train from the original FinBERT weights |
Example β fine-tune the climate head on ESG data:
python -m macro_sentiment.finetune \
--data esg_reports.jsonl \
--text-column body \
--label-column esg_sentiment \
--base-model peyterho/climatebert-macro-sentiment \
--output my-org/climatebert-esg-custom \
--push-to-hub \
--epochs 6 \
--lr 1e-5 \
--max-length 256
Copy-paste these cells into a Colab notebook. Works on the free tier (CPU is fine for small datasets; use a GPU runtime for >5k samples or faster training).
# Clone repo and install dependencies
!git clone https://huggingface.co/peyterho/macro-sentiment-finbert /content/macro-sentiment-finbert
%cd /content/macro-sentiment-finbert
!pip install -q -r requirements.txt evaluate
# Option A: Upload a CSV file via Colab's file browser
from google.colab import files
uploaded = files.upload() # upload your CSV/JSONL file
# Option B: Create a small example dataset inline
import pandas as pd
data = pd.DataFrame({
"text": [
"Markets surged on better-than-expected jobs data",
"Trade war fears sent global equities tumbling",
"Quarterly revenue was roughly in line with guidance",
"Banks rally after stress test results boost confidence",
"Yield curve inversion deepens, sparking recession fears",
"The company maintained its annual dividend outlook",
"Central bank signals patience on future rate decisions",
"Strong retail sales data lifted consumer discretionary stocks",
"Credit downgrades hit emerging market bonds overnight",
"Analysts remain neutral on the sector after mixed earnings",
],
"label": [
"positive", "negative", "neutral", "positive", "negative",
"neutral", "neutral", "positive", "negative", "neutral",
]
})
data.to_csv("my_labels.csv", index=False)
print(f"Created my_labels.csv with {len(data)} rows")
print(data["label"].value_counts())
# Fine-tune on your data (saves locally)
!python -m macro_sentiment.finetune \
--data my_labels.csv \
--text-column text \
--label-column label \
--base-model peyterho/macro-sentiment-finbert \
--output ./my-custom-model \
--epochs 4 \
--lr 2e-5 \
--batch-size 16
from transformers import pipeline
classifier = pipeline("text-classification", model="./my-custom-model", top_k=None)
test_texts = [
"Tech stocks soared to record highs on AI optimism",
"Mounting debt concerns weigh on sovereign credit ratings",
"The index closed flat after a choppy trading session",
]
for text in test_texts:
result = classifier(text)[0]
top = max(result, key=lambda x: x["score"])
print(f'{top["label"]:>8s} ({top["score"]:.2f}) {text}')
from huggingface_hub import login
login() # paste your HF token
# Re-run fine-tuning with push-to-hub, or push the saved model manually:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("./my-custom-model")
tokenizer = AutoTokenizer.from_pretrained("./my-custom-model")
model.push_to_hub("my-username/my-custom-finbert")
tokenizer.push_to_hub("my-username/my-custom-finbert")
print("β
Pushed to Hub!")
If you prefer full control, here's a minimal standalone script that doesn't use finetune.py:
import torch
import numpy as np
from datasets import load_dataset, ClassLabel
from transformers import (
AutoTokenizer,
AutoModelForSequenceClassification,
TrainingArguments,
Trainer,
DataCollatorWithPadding,
)
import evaluate
# 1. Load your data
ds = load_dataset("csv", data_files="my_labels.csv", split="train")
# 2. Map string labels to integers
label_map = {"negative": 0, "neutral": 1, "positive": 2}
ds = ds.map(lambda x: {"label": label_map[x["label"].strip().lower()]})
ds = ds.cast_column("label", ClassLabel(names=["negative", "neutral", "positive"]))
ds = ds.train_test_split(test_size=0.15, seed=42, stratify_by_column="label")
# 3. Load model + tokenizer
model_name = "peyterho/macro-sentiment-finbert"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
# β οΈ FinBERT uses a different label order internally (positive=0, negative=1, neutral=2)
# The finetune.py script handles this automatically via label remapping.
# If writing your own script, remap your unified labels to the model's expected order:
LABEL_REMAP = {0: 1, 1: 2, 2: 0} # unified β FinBERT internal
# 4. Tokenize
def preprocess(examples):
tok = tokenizer(examples["text"], truncation=True, max_length=128)
tok["labels"] = [LABEL_REMAP[l] for l in examples["label"]]
return tok
tokenized = ds.map(preprocess, batched=True, remove_columns=["text", "label"])
# 5. Metrics
acc_metric = evaluate.load("accuracy")
f1_metric = evaluate.load("f1")
def compute_metrics(eval_pred):
logits, labels = eval_pred
preds = np.argmax(logits, axis=-1)
return {
"accuracy": acc_metric.compute(predictions=preds, references=labels)["accuracy"],
"f1_macro": f1_metric.compute(predictions=preds, references=labels, average="macro")["f1"],
}
# 6. Train
training_args = TrainingArguments(
output_dir="./my-finetuned-finbert",
num_train_epochs=4,
per_device_train_batch_size=32,
per_device_eval_batch_size=64,
gradient_accumulation_steps=2,
learning_rate=2e-5,
weight_decay=0.01,
warmup_ratio=0.2,
eval_strategy="epoch",
save_strategy="epoch",
load_best_model_at_end=True,
metric_for_best_model="f1_macro",
bf16=torch.cuda.is_available(),
logging_steps=10,
save_total_limit=2,
)
trainer = Trainer(
model=model,
args=training_args,
train_dataset=tokenized["train"],
eval_dataset=tokenized["test"],
processing_class=tokenizer,
data_collator=DataCollatorWithPadding(tokenizer=tokenizer),
compute_metrics=compute_metrics,
)
trainer.train()
# 7. Evaluate
results = trainer.evaluate()
print(f"Accuracy: {results['eval_accuracy']:.4f}, F1 (macro): {results['eval_f1_macro']:.4f}")
# 8. Save
trainer.save_model("./my-finetuned-finbert")
tokenizer.save_pretrained("./my-finetuned-finbert")
β οΈ Important: Label Remapping
FinBERT's internal label order is
positive=0, negative=1, neutral=2, which differs from the standardnegative=0, neutral=1, positive=2. The includedfinetune.pyscript handles this automatically for all supported base models. If you write your own training script, you must remap labels or your model will learn inverted predictions.The remapping for each base model:
Base model Remap needed? Mapping (unified β model) peyterho/macro-sentiment-finbertβ Yes {0β1, 1β2, 2β0}ProsusAI/finbertβ Yes {0β1, 1β2, 2β0}peyterho/climatebert-macro-sentimentβ Yes {0β2, 1β1, 2β0}peyterho/financial-roberta-large-macro-sentimentβ No Direct mapping soleimanian/financial-roberta-large-sentimentβ No Direct mapping
| Tip | Details |
|---|---|
| Start from this model, not the original FinBERT | peyterho/macro-sentiment-finbert already has macro-financial knowledge baked in. Starting from ProsusAI/finbert throws that away. |
| Use a low learning rate | 1e-5 to 3e-5 works well. Higher rates risk catastrophic forgetting of the pre-trained knowledge. |
| Don't over-train | 2β4 epochs is usually sufficient for fine-tuning. Watch validation loss β if it starts rising, you're overfitting. |
Increase --max-length for long texts |
The default is 128 tokens (~100 words). Set 256 or 512 for analyst reports, earnings transcripts, or policy documents. |
| Class weights handle imbalance | The script automatically computes β(N/nα΅’)-normalized class weights. If your data is balanced, add --no-class-weights. |
| Use the right base for your domain | Climate/ESG text β ClimateBERT head. Policy/macro text β RoBERTa-Large head. General financial β FinBERT (default). |
| Validate on held-out data | The script reserves 15% for validation by default (--val-split 0.15). If you have a dedicated test set, combine everything for training and evaluate separately. |
After fine-tuning, you can swap your custom model into the full MacroSentimentPipeline:
from macro_sentiment import MacroSentimentPipeline
# Override the FinBERT head with your custom model
pipe = MacroSentimentPipeline(
device="cpu",
finbert_model="my-username/my-custom-finbert", # your fine-tuned model
)
# The pipeline will use your model for financial text routing,
# while keeping the other heads (RoBERTa, ClimateBERT, XLM-R) as-is
result = pipe("Your domain-specific financial text here")
print(result.summary())
Or for inference without the full pipeline:
from transformers import pipeline
classifier = pipeline("text-classification", model="my-username/my-custom-finbert", top_k=None)
result = classifier("Your domain-specific text here")
print(result)
mode="all" for important decisions to get ensemble averaging across all heads.If you use this model or pipeline, please cite the base models:
@article{araci2019finbert,
title={FinBERT: Financial Sentiment Analysis with Pre-Trained Language Models},
author={Araci, Dogu},
journal={arXiv preprint arXiv:1908.10063},
year={2019}
}
@article{loughran2011liability,
title={When is a liability not a liability? Textual analysis, dictionaries, and 10-Ks},
author={Loughran, Tim and McDonald, Bill},
journal={The Journal of Finance},
volume={66},
number={1},
pages={35--65},
year={2011}
}
@article{henry2008earnings,
title={Are investors influenced by how earnings press releases are written?},
author={Henry, Elaine},
journal={Journal of Business Communication},
volume={45},
number={4},
pages={363--407},
year={2008}
}
βββ model.safetensors # FinBERT fine-tuned weights (109M params)
βββ config.json # Model config (3-class: positive/negative/neutral)
βββ tokenizer.json # BERT tokenizer
βββ eval_results.json # In-domain benchmarks for all heads
βββ eval_ood_results.json # Out-of-domain benchmarks
βββ artifacts/
β βββ meta_classifier.joblib # GradientBoosting meta-classifier on dictionary features
β βββ smart_router.joblib # Topic routing model
βββ macro_sentiment/
β βββ __init__.py # Public API exports
β βββ pipeline.py # MacroSentimentPipeline β main entry point
β βββ transformers_ensemble.py # Multi-head transformer ensemble + topic router
β βββ dictionaries.py # LM, Henry, Climate, Macro dictionary scorers
β βββ data_prep.py # Dataset loading and combination (5 datasets)
β βββ finetune.py # Custom fine-tuning script
β βββ train_meta.py # Meta-classifier training script
βββ requirements.txt