Transformer课程第8课 NER案例模型训练及预测_如何用transformer训练ner数据

作者：盐析白兔 | 2024-04-30 14:51:27

踩

如何用transformer训练ner数据

Transformer课程第8课 NER案例模型训练及预测

Train Our Classification Model

现在，我们的输入数据已正确格式化，是时候对BERT模型进行微调了。

查询标签的数量为17，所有标签加上填充标签的数量是18

# This is the number of labels in our dataset, so we will be doing 18-way classification (all labels plus our padding label)
len(label_map)
1
2

加载BertForTokenClassification

from transformers import BertForTokenClassification, AdamW, BertConfig

# Load B1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/513876