Views
No views yet
ProsusAI/finbert model fine-tuned for Named Entity Recognition (NER) to detect company names in financial news articles. Trained on Indian financial news but generalises to broader financial English text.ProsusAI/finbert (110M params, BERT pre-trained on financial text)O, B-COMPANY, I-COMPANY| Metric | Entity-level | Token-level |
|---|---|---|
| Precision | 0.43 | 0.67 |
| Recall | 0.58 | 0.88 |
| F1 | 0.49 | 0.76 |
| Pred O | Pred COMPANY | |
|---|---|---|
| True O | 76,046 | 948 |
| True COMPANY | 262 | 1,910 |
1from transformers import pipeline
2
3nlp = pipeline(
4 "ner",
5 model="ritam-m/finbert-company-ner",
6 aggregation_strategy="first",
7)
8
9text = "HDFC Bank reported strong Q1 earnings, while Infosys maintained its FY27 guidance."
10results = nlp(text)
11
12for entity in results:
13 if entity["entity_group"] == "COMPANY":
14 print(entity["word"], entity["score"])ritam-m/bert-base-company-ner which trades recall for precision