京公网安备 11010802034615号
经营许可证编号:京B2-20210330
自然语言处理(NLP)是人工智能领域中一个快速发展的分支,它提供了许多技术和方法来对自然语言进行处理。其中,词嵌入(word embedding)是NLP中最重要的技术之一,因为它允许将自然语言转换为计算机可以理解和处理的向量表示形式。
BERT(Bidirectional Encoder Representations from Transformers)是一种有监督的预训练模型,它使用了Transformer架构,并在大型语料库上进行了训练,可以用于各种自然语言处理任务,如文本分类、句子配对等。
BERT模型的输出包含多个层级,其中第一层是输入层,最后一层是输出层,而在中间的隐藏层中,每一个单词都被映射到一个低维度的向量空间中。这些向量就是所谓的BERT词嵌入。
提取BERT词嵌入非常简单,只需要将文本输入BERT模型中,并获取相应隐藏层的输出即可。具体步骤如下:
首先,我们需要安装相应的Python库,包括transformers和torch。可以使用以下命令来安装这些库:
!pip install transformers
!pip install torch
接下来,加载BERT模型并设置为评估模式,以保证Dropout和BatchNormalization层不会被激活。我们可以使用以下代码完成这一步骤:
from transformers import BertTokenizer, BertModel
# 加载BertTokenizer和BertModel
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
model = BertModel.from_pretrained('bert-base-uncased')
# 设置为评估模式
model.eval()
然后,我们需要将文本转换为BERT可接受的输入格式。具体来说,我们需要使用BertTokenizer对文本进行分词,并将结果转换为BERT的输入ID和Attention Mask张量。以下是一个示例代码:
text = "I love natural language processing."
tokens = tokenizer.tokenize(text)
input_ids = torch.tensor([tokenizer.convert_tokens_to_ids(tokens)])
attention_mask = torch.ones_like(input_ids)
最后,我们可以将输入张量传递给BERT模型并获取相应的隐藏层输出。具体来说,我们将输入ID和Attention Mask张量传递给BertModel,并获取相应的所有隐藏层输出。以下是一个示例代码:
with torch.no_grad():
outputs = model(input_ids, attention_mask=attention_mask)
hidden_states = outputs[2]
在此示例中,我们获取了BERT模型的所有隐藏层输出,可以根据需要选择其中任意一层作为词嵌入。
总之,BERT是一种非常强大的预训练模型,可以用于各种自然语言处理任务。它的词嵌入提取非常简单,只需要将文本输入BERT模型中,并获取相应隐藏层的输出即可。
数据分析咨询请扫描二维码
若不方便扫码,搜微信号:CDAshujufenxi
核心资源推荐 CDA教材(2025):https://edu.cda.cn/goods/show/3151CDA模拟题(2025):https://edu.cda.cn/goods/show/2845CD ...
2026-10-26核心资源推荐 CDA教材(2025):https://edu.cda.cn/goods/show/3151CDA模拟题(2025):https://edu.cda.cn/goods/show/2845CD ...
2026-10-26运营岗的技能提升,最容易走偏的一步是先去学工具。Excel、SQL、Python 都值得学,但如果不知道自己要回答什么业务问题、指标口 ...
2026-10-26营销人提升业务能力,最常见的误区是“缺什么补什么”:文案不行学写作,投放不行学出价,活动效果差就再学一套裂变玩法。零散技 ...
2026-10-26营销经理的能力提升,最容易走成两条弯路:一条是零散学工具,Excel、SQL、Python挨个报课,学完却不知道用在哪;另一条是把“提 ...
2026-10-26营销经理的沟通难点,很少是“不会说话”。更多时候是三类具体卡点:向管理层汇报,讲了十分钟还没说到结论;向产品、财务、销售 ...
2026-10-26营销岗位的能力提升,最容易走偏的一条路是先学工具。Excel、SQL、Python排着队学,学完却发现日常工作中还是说不清一场活动到底 ...
2026-10-26营销岗日常要回答的问题其实很集中:这次投放带来多少有效线索,活动哪一步掉得最多,新客和老客谁在复购,渠道花出去的钱能不能 ...
2026-10-26营销策划问“最应该掌握什么数据分析技能”,答案不是先学某个工具,而是先掌握把活动目标拆成指标与口径、再用数据评估活动效果 ...
2026-10-26信贷专员的培训课不少,难的是选课顺序。常见误区是看到“数据分析”“风险建模”就报名,学完却发现与手头工作接不上;也有人反 ...
2026-10-26信贷专员的核心工作,是把一笔贷款从客户申请一路管到结清:贷前判断客户还得起吗,贷中盯住条件有没有落实、风险有没有变化,贷 ...
2026-10-26新媒体运营提升业务技能,不建议从“多学几个工具”开始。更有效的做法是先搭一个能力框架,再用岗位任务反推短板。可以按内容策 ...
2026-10-26营销岗位的沟通问题,很少是“嘴笨”。更多时候是三类信息没对齐:对面是谁、这次沟通要对方做什么、凭什么相信你。同一套话术, ...
2026-10-26一、SmartArt 是什么,为什么用它画流程图 在 Excel 里画流程图,很多人第一反应是插入"形状"——矩形、箭头、连线一个个手动画 ...
2026-10-26税务专员新人想快速上手,最有效的路径不是先背政策条文,而是先把日常任务跑通:发票台账能登记清楚、申报表能和账面数据对上、 ...
2026-10-26一、什么是聚类分析,什么时候用它 聚类分析是一种无监督学习方法,它把一批对象按照"相似性"自动分成若干组,让同一组内的对象 ...
2026-10-26税务专员日常要面对申报表、发票明细、税负台账和各类核对底稿,常用的数据分析模型其实集中在六类:对比分析、趋势分析、结构分 ...
2026-10-26数字营销专员提升知识,最有效的途径不是零散刷课、追热点工具,而是围绕岗位任务搭一张“渠道—内容—用户—数据”的知识骨架, ...
2026-10-26数字营销不是一个只靠创意和网感的岗位。招聘方在JD里写“熟悉投放”“会看数据”“能做用户增长”,背后其实是一套可以拆开的知 ...
2026-10-26先给结论:做业务分析、数据运营这类岗位,不需要先啃高数和线性代数,但描述统计、图表阅读和指标口径必须掌握;只有走建模、挖 ...
2026-10-26