tensorflow - 在哪里可以找到预训练的 SparkNLP NerDLModel 的类标签列表？

Question

我已经搜索了一段时间，但没能找到预训练的 NerDL(tensorflow) 模型中包含哪些 NER 标签。我认为训练数据可以提供这样的信息，但我没有看到任何文档中提到它。

可下载模型: https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/ner_precise_en_1.7.0_2_1539623388047.zip

任何方向将不胜感激!

更新:

我确实按照这里的建议在 SparkNLP github 中提交了一个问题 :)我刚收到他们的回音。这是答案:

For practical purposes, the pretrained NER model has

B-ORG

I-ORG

B-PER

I-PER

B-LOC

I-LOC

and it has been trained from:https://raw.githubusercontent.com/patverga/torch-ner-nlp-from-scratch/master/data/conll2003/eng.train

参见原始问题 here .

Answer 1

该模型是在 NER 的 CONLL2003 数据集上训练的，

http://aclweb.org/anthology/W03-0419

该数据集基本上包含 PERSON、ORGANIZATION 和 LOCATION。

希望对您有所帮助，阿尔贝托。