Bert Base Hungarian Cased Ner

🚀 匈牙利命名實體識別 (NER) 模型

本模型是基於“SZTAKI-HLT/hubert-base-cc”模型，使用著名的WikiANN數據集進行微調得到的。該數據集在論文“Cross-lingual Name Tagging and Linking for 282 Languages” 點擊查看 中有詳細介紹。

🚀 快速開始

微調參數

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 

使用方法

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>")

有關使用 aggregation_strategy 參數進行實體分組的詳細信息，請參考 https://huggingface.co/transformers/_modules/transformers/pipelines/token_classification.html。

📊 參考測試結果

• 準確率：0.9774538310923768
• F1值：0.9462099085573904
• 精確率：0.9425718667406271
• 召回率：0.9498761426661113

📄 許可證

本項目採用MIT許可證。