许可证: cc-by-nc-4.0
数据集:

• Universal-NER/Pile-NER-type
  管道标签: token-classification
  库名称: gliner

GLiNER-small 模型卡

GLiNER 是一个命名实体识别（NER）模型，能够使用双向Transformer编码器（类似BERT）识别任何实体类型。它为传统NER模型提供了实用替代方案——后者受限于预定义实体，同时也为大型语言模型（LLMs）提供了轻量化选择——尽管LLMs具有灵活性，但在资源受限场景下成本高昂且体积庞大。

本版本基于Pile-NER数据集训练（研究用途）

相关链接

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

可用模型

安装

使用前需安装GLiNER库：

!pip install gliner

使用示例

加载模型后，可通过predict_entities函数预测实体：

from gliner import GLiNER  

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

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日 => 日期  
阿尔纳斯尔 => 球队  
葡萄牙国家男子足球队 => 球队  
金球奖 => 奖项  
欧足联年度最佳男子球员奖 => 奖项  
欧洲金靴奖 => 奖项  
欧冠联赛 => 赛事  
欧洲杯 => 赛事  
欧国联 => 赛事  

NER基准测试结果

模型作者

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