NER_on_Bangla_Language

engineersakibcse47/NER_on_Bangla_Language

NER_on_Bangla_Language is a token classification model from engineersakibcse47. Use it when you need labels on individual words, such as names. It is set up for transformers.

<h1Named Entity Recognition on Bangla Language</h1 Fine Tuning BERT for NER on Bengali Language Tagging using HuggingFace

Downloads · 30 days

61% of all-time downloads

All-time downloads

Public

Parameters

177M

2.1 GB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.pt1.4 GB · 67%

At a glance

Task: Token Classification
Library: transformers
Model type: bert
Access: Public
Created: May 10, 2024
Updated: May 16, 2024
SHA: d97e5d44

Try a prompt

Datasets

wikiann

Model card

From the Hugging Face model README

<h1>Named Entity Recognition on Bangla Language</h1> Fine Tuning BERT for NER on Bengali Language Tagging using HuggingFace

Correspondence Label ID and Label Name

Label ID	Label Name
0	O
1	B-PER
2	I-PER
3	B-ORG
4	I-ORG
5	B-LOC
6	I-LOC

<h1>Evaluation and Validation</h1>

Name	Precision	Recall	F1	Accuracy
Train/Val set	0.963899	0.964770	0.964334	0.981252
Test set	0.952855	0.965105	0.958941	0.981349

Transformers AutoModelForTokenClassification

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("engineersakibcse47/NER_on_Bangla_Language")
model_ner = AutoModelForTokenClassification.from_pretrained("engineersakibcse47/NER_on_Bangla_Language")

pipe = pipeline("ner", model=model_ner, tokenizer=tokenizer, aggregation_strategy="simple")

sample = "বসনিয়া ও হার্জেগোভিনা"

result = pipe(sample)
result

Task: Token Classification
Library: transformers
Type: bert
Languages: bn
Created: May 10, 2024
Updated: May 16, 2024