Bert Base Arabic Camelbert Mix Ner

🚀 CAMeLBERT-Mix NER模型

CAMeLBERT-Mix NER模型是一個命名實體識別（NER）模型，旨在解決阿拉伯語命名實體識別的問題，為阿拉伯語自然語言處理提供了有效的工具，提升了相關任務的處理效率和準確性。

🚀 快速開始

你可以直接將CAMeLBERT-Mix NER模型作為我們 CAMeL Tools NER組件的一部分使用（推薦），也可以將其作為transformers管道的一部分使用。

✨ 主要特性

• 基於預訓練模型微調：該模型通過對 CAMeLBERT Mix 模型進行微調構建而成。
• 使用特定數據集：在微調過程中，使用了 ANERcorp 數據集。
• 有詳細研究支持：微調過程和使用的超參數可在論文 "The Interplay of Variant, Size, and Task Type in Arabic Pre-trained Language Models" 中找到，微調代碼可在 此處 獲取。

💻 使用示例

基礎用法

若要將該模型與 CAMeL Tools NER組件一起使用：

>>> from camel_tools.ner import NERecognizer
>>> from camel_tools.tokenizers.word import simple_word_tokenize
>>> ner = NERecognizer('CAMeL-Lab/bert-base-arabic-camelbert-mix-ner')
>>> sentence = simple_word_tokenize('إمارة أبوظبي هي إحدى إمارات دولة الإمارات العربية المتحدة السبع')
>>> ner.predict_sentence(sentence)
>>> ['O', 'B-LOC', 'O', 'O', 'O', 'O', 'B-LOC', 'I-LOC', 'I-LOC', 'O']

高級用法

你也可以直接將NER模型與transformers管道一起使用：

>>> from transformers import pipeline
>>> ner = pipeline('ner', model='CAMeL-Lab/bert-base-arabic-camelbert-mix-ner')
>>> ner("إمارة أبوظبي هي إحدى إمارات دولة الإمارات العربية المتحدة السبع")
[{'word': 'أبوظبي',
  'score': 0.9895730018615723,
  'entity': 'B-LOC',
  'index': 2,
  'start': 6,
  'end': 12},
 {'word': 'الإمارات',
  'score': 0.8156259655952454,
  'entity': 'B-LOC',
  'index': 8,
  'start': 33,
  'end': 41},
 {'word': 'العربية',
  'score': 0.890906810760498,
  'entity': 'I-LOC',
  'index': 9,
  'start': 42,
  'end': 49},
 {'word': 'المتحدة',
  'score': 0.8169114589691162,
  'entity': 'I-LOC',
  'index': 10,
  'start': 50,
  'end': 57}]

⚠️ 重要提示

要下載我們的模型，你需要 transformers>=3.5.0。否則，你可以手動下載模型。

📚 詳細文檔

模型描述

CAMeLBERT-Mix NER模型 是一個命名實體識別（NER）模型，它通過對 CAMeL-Lab/bert-base-arabic-camelbert-mix 模型進行微調構建而成。在微調過程中，使用了 ANERcorp 數據集。我們的微調過程和使用的超參數可在論文 "The Interplay of Variant, Size, and Task Type in Arabic Pre-trained Language Models" 中找到，微調代碼可在 此處 獲取。

預期用途

你可以直接將CAMeLBERT-Mix NER模型作為我們 CAMeL Tools NER組件的一部分使用（推薦），也可以將其作為transformers管道的一部分使用。

📄 許可證

本項目採用Apache-2.0許可證。

📚 引用

如果你使用了該模型，請引用以下論文：

@inproceedings{inoue-etal-2021-interplay,
    title = "The Interplay of Variant, Size, and Task Type in {A}rabic Pre-trained Language Models",
    author = "Inoue, Go  and
      Alhafni, Bashar  and
      Baimukan, Nurpeiis  and
      Bouamor, Houda  and
      Habash, Nizar",
    booktitle = "Proceedings of the Sixth Arabic Natural Language Processing Workshop",
    month = apr,
    year = "2021",
    address = "Kyiv, Ukraine (Online)",
    publisher = "Association for Computational Linguistics",
    abstract = "In this paper, we explore the effects of language variants, data sizes, and fine-tuning task types in Arabic pre-trained language models. To do so, we build three pre-trained language models across three variants of Arabic: Modern Standard Arabic (MSA), dialectal Arabic, and classical Arabic, in addition to a fourth language model which is pre-trained on a mix of the three. We also examine the importance of pre-training data size by building additional models that are pre-trained on a scaled-down set of the MSA variant. We compare our different models to each other, as well as to eight publicly available models by fine-tuning them on five NLP tasks spanning 12 datasets. Our results suggest that the variant proximity of pre-training data to fine-tuning data is more important than the pre-training data size. We exploit this insight in defining an optimized system selection model for the studied tasks.",
}