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

时间:2018-11-26 19:13:06

标签: tensorflow johnsnowlabs-spark-nlp

我已经搜索了一段时间,但没有运气发现预训练的NerDL(tensorflow)模型中包含哪些NER标签。我认为培训数据可以提供此类信息,但是我看不到任何文档中提到的内容。

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

任何方向将不胜感激!

更新:

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

  

出于实际目的,预训练的NER模型具有

     

B-ORG

     

I-ORG

     

B-PER

     

I-PER

     

B-LOC

     

I-LOC

     

,它的训练来源为:   https://raw.githubusercontent.com/patverga/torch-ner-nlp-from-scratch/master/data/conll2003/eng.train

请参见原始问题here

1 个答案:

答案 0 :(得分:0)

该模型在CONLL2003数据集上针对NER进行了训练,

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

该数据集基本上具有PERSON,ORGANIZATION和LOCATION。

希望这会有所帮助, 阿尔贝托。

相关问题