Bert Base Hungarian Cased Ner

🚀 Hungarian Named Entity Recognition (NER) Model

This model is a fine - tuned version of "SZTAKI - HLT/hubert - base - cc", leveraging the well - known WikiANN dataset. It is designed to perform named entity recognition in Hungarian.

🚀 Quick Start

This model is the fine - tuned model of "SZTAKI - HLT/hubert - base - cc" using the famous WikiANN dataset presented in the "Cross - lingual Name Tagging and Linking for 282 Languages" paper.

🔧 Technical Details

Fine - tuning parameters:

task = "ner"
model_checkpoint = "SZTAKI - HLT/hubert - base - cc"
batch_size = 8 
label_list = ['O', 'B - PER', 'I - PER', 'B - ORG', 'I - ORG', 'B - LOC', 'I - LOC']
max_length = 512 
learning_rate = 2e - 5 
num_train_epochs = 3 
weight_decay = 0.01 

💻 Usage Examples

Basic Usage

model = AutoModelForTokenClassification.from_pretrained("akdeniz27/bert - base - hungarian - cased - ner")
tokenizer = AutoTokenizer.from_pretrained("akdeniz27/bert - base - hungarian - cased - ner")
ner = pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy="first")
ner("<your text here>")

Pls refer "https://huggingface.co/transformers/_modules/transformers/pipelines/token_classification.html" for entity grouping with aggregation_strategy parameter.

📚 Documentation

Reference test results:

📄 License

This project is licensed under the MIT License.