我已经搜索了一段时间,但没有运气发现预训练的NerDL(tensorflow)模型中包含哪些NER标签。我认为培训数据可以提供此类信息,但是我看不到任何文档中提到的内容。
任何方向将不胜感激!
更新:
我确实按照这里的建议在SparkNLP github中提出了一个问题:) 我只是听到他们的回音。答案是这样:
出于实际目的,预训练的NER模型具有
B-ORG
I-ORG
B-PER
I-PER
B-LOC
I-LOC
请参见原始问题here。
答案 0 :(得分:0)
该模型在CONLL2003数据集上针对NER进行了训练,
http://aclweb.org/anthology/W03-0419
该数据集基本上具有PERSON,ORGANIZATION和LOCATION。
希望这会有所帮助, 阿尔贝托。