您当前的位置: 首页 > 

段智华

暂无认证

  • 0浏览

    0关注

    1232博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

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

段智华 发布时间:2021-12-01 21:34:53 ,浏览量:0

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

Train Our Classification Model

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

4.1. BertForTokenClassification

查询标签的数量为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)

17

加载BertForTokenClassification

from transformers import BertForTokenClassification, AdamW, BertConfig

# Load B
关注
打赏
1659361485
查看更多评论
立即登录/注册

微信扫码登录

0.4062s