带有Hugginface Tensorflow 2的NER用BERT

时间:2020-02-25 00:51:06

标签: python tensorflow nlp text-classification ner

在Tensorflow中将BERT用于NER时,我需要一些帮助。基本上所有教程都在PyTorch中。我真的很喜欢这两种方法中的一些建议。

1)我对使用Huggingface Transformers软件包感兴趣,因为它们具有TFBertForTokenClassification类,但是没有有关如何使用它的教程或指南。也没有如何训练自己的数据集。

2)我在Tensorflow中实现了BERT,以进行文本分类,除了原始bert库之外,没有其他API /库。一种方法是使它适应NER,但是我不确定如何更改模型。目前,我只是在预训练的BERT模型的顶部抛出了一个辍学层和一个密集层,其中softmax遍历了该密集层。

0 个答案:

没有答案