license: cc-by-nc-4.0 language:

• en pipeline_tag: token-classification datasets:
• Universal-NER/Pile-NER-type library_name: gliner

GLiNER-base 模型卡

GLiNER 是一个命名实体识别（NER）模型，能够通过双向 Transformer 编码器（类似 BERT）识别任何类型的实体。它为传统 NER 模型提供了实用替代方案——传统模型局限于预定义实体类型，而大型语言模型（LLMs）虽然灵活，但在资源受限场景下成本高昂且体积庞大。

相关链接

• 论文: https://arxiv.org/abs/2311.08526
• 代码库: https://github.com/urchade/GLiNER

可用模型

安装

使用本模型需先安装 GLiNER Python 库：

!pip install gliner

使用方式

安装 GLiNER 库后，导入 GLiNER 类即可加载本模型。通过 GLiNER.from_pretrained 加载模型后，可使用 predict_entities 进行实体预测。

from gliner import GLiNER

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

text = """
克里斯蒂亚诺·罗纳尔多·多斯·桑托斯·阿维罗（葡萄牙语发音：[kɾiʃˈtjɐnu ʁɔˈnaldu]；1985年2月5日出生）是葡萄牙职业足球运动员，司职前锋，同时担任沙特职业足球联赛阿尔纳斯尔俱乐部和葡萄牙国家队的双料队长。被广泛认为是有史以来最伟大的球员之一，罗纳尔多曾五次获得金球奖[注3]，创纪录地三次荣获欧足联年度最佳男子球员奖，以及四次欧洲金靴奖，是欧洲球员中获得该奖项次数最多的选手。其职业生涯共赢得33座冠军奖杯，包括7次联赛冠军、5次欧冠联赛冠军、1次欧洲杯冠军和1次欧国联冠军。罗纳尔多保持着欧冠联赛出场次数（183场）、进球数（140球）和助攻数（42次）的纪录，同时是欧洲杯进球数（14球）、国家队进球数（128球）和国家队出场次数（205场）的纪录保持者。他是少数职业生涯出场超过1200场的球员之一（非门将位置球员之最），并为俱乐部和国家队攻入超过850粒正式比赛进球，成为历史最佳射手。
"""

labels = ["人物", "奖项", "日期", "赛事", "球队"]

entities = model.predict_entities(text, labels)

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

克里斯蒂亚诺·罗纳尔多·多斯·桑托斯·阿维罗 => 人物
1985年2月5日 => 日期
阿尔纳斯尔 => 球队
葡萄牙国家队 => 球队
金球奖 => 奖项
欧足联年度最佳男子球员奖 => 奖项
欧洲金靴奖 => 奖项
欧冠联赛 => 赛事
欧洲杯 => 赛事
欧国联 => 赛事
欧冠联赛 => 赛事
欧洲杯 => 赛事

命名实体识别基准测试结果

模型作者

模型作者包括：

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

引用文献

@misc{zaratiana2023gliner,
      title={GLiNER: 基于双向Transformer的通用命名实体识别模型}, 
      author={Urchade Zaratiana and Nadi Tomeh and Pierre Holat and Thierry Charnois},
      year={2023},
      eprint={2311.08526},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}