Gliner Small V2.1

🚀 GLiNER

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

🚀 クイックスタート

GLiNERを使用するには、まずGLiNER Pythonライブラリをインストールする必要があります。その後、モデルをロードして固有表現を予測することができます。

✨ 主な機能

• 双方向トランスフォーマーエンコーダを使用して、任意のエンティティタイプを識別できます。
• 従来のNERモデルや大規模言語モデルに代わる実用的な選択肢を提供します。

📦 インストール

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

!pip install gliner

💻 使用例

基本的な使用法

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

from gliner import GLiNER

model = GLiNER.from_pretrained("urchade/gliner_small-v2.1")

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

📚 ドキュメント

リンク

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

利用可能なモデル

固有表現抽出のベンチマーク結果

モデルの作成者

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

• 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}
}

📄 ライセンス

このモデルは、Apache-2.0ライセンスの下で提供されています。