🚀 荷兰语命名实体识别(Flair默认模型)
这是 Flair 库中用于荷兰语的标准4类命名实体识别(NER)模型。该模型在CoNLL - 03数据集上的F1分数达到了 92.58。它可以预测4种标签,具体如下:
标签 |
含义 |
PER |
人名 |
LOC |
地名 |
ORG |
组织名 |
MISC |
其他名称 |
该模型基于Transformer嵌入和LSTM - CRF构建。
🚀 快速开始
依赖安装
本模型需要安装 Flair 库,可以使用以下命令进行安装:
pip install flair
代码示例
from flair.data import Sentence
from flair.models import SequenceTagger
tagger = SequenceTagger.load("flair/ner-dutch")
sentence = Sentence("George Washington ging naar Washington")
tagger.predict(sentence)
print(sentence)
print('The following NER tags are found:')
for entity in sentence.get_spans('ner'):
print(entity)
输出示例
运行上述代码会得到如下输出:
Span [1,2]: "George Washington" [− Labels: PER (0.997)]
Span [5]: "Washington" [− Labels: LOC (0.9996)]
这表明在句子 "George Washington ging naar Washington" 中,识别出了实体 "George Washington"(标签为 人名)和 "Washington"(标签为 地名)。
💻 使用示例
基础用法
上述代码展示了如何在Flair中使用该模型进行荷兰语命名实体识别,主要步骤包括加载模型、创建句子、进行预测和输出结果。
高级用法
以下是训练该模型的脚本示例:
from flair.data import Corpus
from flair.datasets import CONLL_03_DUTCH
from flair.embeddings import WordEmbeddings, StackedEmbeddings, FlairEmbeddings
corpus: Corpus = CONLL_03_DUTCH()
tag_type = 'ner'
tag_dictionary = corpus.make_tag_dictionary(tag_type=tag_type)
embeddings = TransformerWordEmbeddings('wietsedv/bert-base-dutch-cased')
tagger: SequenceTagger = SequenceTagger(hidden_size=256,
embeddings=embeddings,
tag_dictionary=tag_dictionary,
tag_type=tag_type)
trainer: ModelTrainer = ModelTrainer(tagger, corpus)
trainer.train('resources/taggers/ner-dutch',
train_with_dev=True,
max_epochs=150)
📄 引用说明
使用此模型时,请引用以下论文:
@inproceedings{akbik-etal-2019-flair,
title = "{FLAIR}: An Easy-to-Use Framework for State-of-the-Art {NLP}",
author = "Akbik, Alan and
Bergmann, Tanja and
Blythe, Duncan and
Rasul, Kashif and
Schweter, Stefan and
Vollgraf, Roland",
booktitle = "Proceedings of the 2019 Conference of the North {A}merican Chapter of the Association for Computational Linguistics (Demonstrations)",
year = "2019",
url = "https://www.aclweb.org/anthology/N19-4010",
pages = "54--59",
}
🔗 问题反馈
如果你在使用过程中遇到问题,可以在 Flair问题跟踪器 中反馈。