Text Classification
Transformers
Safetensors
English
bert
finbert
finance
sentiment
sentiment-analysis
financial-sentiment
text-embeddings-inference
Instructions to use ENTUM-AI/FinBERT-Pro with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ENTUM-AI/FinBERT-Pro with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ENTUM-AI/FinBERT-Pro")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("ENTUM-AI/FinBERT-Pro") model = AutoModelForSequenceClassification.from_pretrained("ENTUM-AI/FinBERT-Pro", device_map="auto") - Notebooks
- Google Colab
- Kaggle
metadata
language:
- en
license: apache-2.0
library_name: transformers
tags:
- text-classification
- bert
- finbert
- finance
- sentiment
- sentiment-analysis
- financial-sentiment
datasets:
- FinanceInc/auditor_sentiment
- nickmuchi/financial-classification
- warwickai/financial_phrasebank_mirror
pipeline_tag: text-classification
π― FinBERT-Pro
An improved financial sentiment model built on ProsusAI/finbert. Fine-tuned on 3 expert-annotated financial datasets for more robust sentiment classification.
The model provides softmax outputs for three sentiment classes: Positive, Negative, Neutral.
π Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="ENTUM-AI/FinBERT-Pro")
classifier("Stock price soars on record-breaking earnings report")
# [{'label': 'Positive', 'score': 0.99}]
classifier("Company announces quarterly earnings results")
# [{'label': 'Neutral', 'score': 0.98}]
classifier("Revenue decline signals weakening market position")
# [{'label': 'Negative', 'score': 0.98}]
π Training Data
Fine-tuned on 3 expert-annotated public datasets:
| Dataset | Samples |
|---|---|
| FinanceInc/auditor_sentiment | ~4.8K |
| nickmuchi/financial-classification | ~5K |
| warwickai/financial_phrasebank_mirror | ~4.8K |
Unlike the original FinBERT (trained on a single dataset), FinBERT-Pro combines multiple expert-annotated sources for better generalization across different financial text styles.
π What's Different from FinBERT?
- Multiple data sources β trained on 3 expert-annotated datasets instead of 1
- Class-weighted training β handles imbalanced label distributions
- Better generalization β diverse training data improves robustness on unseen financial texts
β οΈ Limitations
- English only
- Designed for short financial texts (headlines, news, reports)