Bert完形填空

Shy960418

于 2024-11-06 10:36:34 发布

阅读量532

点赞数 2

分类专栏：深度学习文章标签：深度学习 bert python

本文链接：https://blog.csdn.net/m0_37134868/article/details/143562443

版权

转载自：| 03_language_model/03_Bert完形填空.ipynb | 基于transformers使用Bert模型做完形填空 |Open In Colab |

完形填空

利用语言模型，可以完成完形填空（fill mask），预测缺失的单词。
当前，效果最好的语言模型是Bert系列的预训练语言模型。

!pip install transformers

import os

from transformers import pipeline

os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
model_name = "hfl/chinese-macbert-base"

nlp = pipeline("fill-mask",
               model=model_name,
               tokenizer=model_name,
               device=-1,

最低0.47元/天解锁文章