Gliner Base

🚀 GLiNER-base モデルカード

GLiNERは、双方向トランスフォーマーエンコーダ（BERTライク）を使用して任意のエンティティタイプを識別できる固有表現認識（NER）モデルです。これは、事前定義されたエンティティに限定される従来のNERモデルや、柔軟性はあるもののリソース制約のあるシナリオではコストが高く大規模な大規模言語モデル（LLM）に代わる実用的な選択肢を提供します。

🚀 クイックスタート

GLiNERは、双方向トランスフォーマーエンコーダを用いた固有表現認識モデルです。以下の手順で使用を開始できます。

🔗 リンク

• 論文: https://arxiv.org/abs/2311.08526
• リポジトリ: https://github.com/urchade/GLiNER

✨ 主な機能

GLiNERは、任意のエンティティタイプを識別できるため、従来のNERモデルや大規模言語モデルに代わる実用的な選択肢を提供します。

📦 インストール

このモデルを使用するには、GLiNERのPythonライブラリをインストールする必要があります。

!pip install gliner

💻 使用例

基本的な使用法

GLiNERライブラリをダウンロードしたら、GLiNERクラスをインポートできます。次に、GLiNER.from_pretrainedを使用してこのモデルをロードし、predict_entitiesでエンティティを予測できます。

from gliner import GLiNER

model = GLiNER.from_pretrained("urchade/gliner_base")

text = """
Cristiano Ronaldo dos Santos Aveiro (Portuguese pronunciation: [kɾiʃˈtjɐnu ʁɔˈnaldu]; born 5 February 1985) is a Portuguese professional footballer who plays as a forward for and captains both Saudi Pro League club Al Nassr and the Portugal national team. Widely regarded as one of the greatest players of all time, Ronaldo has won five Ballon d'Or awards,[note 3] a record three UEFA Men's Player of the Year Awards, and four European Golden Shoes, the most by a European player. He has won 33 trophies in his career, including seven league titles, five UEFA Champions Leagues, the UEFA European Championship and the UEFA Nations League. Ronaldo holds the records for most appearances (183), goals (140) and assists (42) in the Champions League, goals in the European Championship (14), international goals (128) and international appearances (205). He is one of the few players to have made over 1,200 professional career appearances, the most by an outfield player, and has scored over 850 official senior career goals for club and country, making him the top goalscorer of all time.
"""

labels = ["person", "award", "date", "competitions", "teams"]

entities = model.predict_entities(text, labels)

for entity in entities:
    print(entity["text"], "=>", entity["label"])

Cristiano Ronaldo dos Santos Aveiro => person
5 February 1985 => date
Al Nassr => teams
Portugal national team => teams
Ballon d'Or => award
UEFA Men's Player of the Year Awards => award
European Golden Shoes => award
UEFA Champions Leagues => competitions
UEFA European Championship => competitions
UEFA Nations League => competitions
Champions League => competitions
European Championship => competitions

📚 ドキュメント

固有表現認識のベンチマーク結果

モデルの作成者

このモデルの作成者は以下の通りです。

• Urchade Zaratiana
• Nadi Tomeh
• Pierre Holat
• Thierry Charnois

引用

@misc{zaratiana2023gliner,
      title={GLiNER: Generalist Model for Named Entity Recognition using Bidirectional Transformer}, 
      author={Urchade Zaratiana and Nadi Tomeh and Pierre Holat and Thierry Charnois},
      year={2023},
      eprint={2311.08526},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

📄 ライセンス

このモデルは、cc-by-nc-4.0またはapache-2.0ライセンスの下で提供されています。