在Tensorflow中将BERT用于NER时,我需要一些帮助。基本上所有教程都在PyTorch中。我真的很喜欢这两种方法中的一些建议。
1)我对使用Huggingface Transformers软件包感兴趣,因为它们具有TFBertForTokenClassification
类,但是没有有关如何使用它的教程或指南。也没有如何训练自己的数据集。
2)我在Tensorflow中实现了BERT,以进行文本分类,除了原始bert
库之外,没有其他API /库。一种方法是使它适应NER,但是我不确定如何更改模型。目前,我只是在预训练的BERT模型的顶部抛出了一个辍学层和一个密集层,其中softmax遍历了该密集层。