京公网安备 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数据分析师 出品 作者:李诗怡 四层结构 · 零件与房子 · 六大易错点 一、什么是指标体系 定义:指标体系是整套完整框架 ...
2026-10-13写在开头 先还原一个你大概率见过的场面。 立项会上有人说:我们得做用户画像。 于是团队关进会议室,白板写满便利贴,两小时后 ...
2026-10-13 很多数据分析师每天都在计算指标、制作报表,但当被问到“什么叫指标数据元”“指标数据标准包含哪些核心维度”“指标数据质 ...
2026-10-13在统计学与数据分析研究中,相关性分析分为数值变量分析与分类变量分析两大类。针对身高、销量、分数等连续数值变量,通常采用皮 ...
2026-10-12在Power BI可视化分析体系中,常规表格仅能展示枯燥的数值数据,存在数据对比不直观、强弱规律不突出、维度交叉分析低效等问题。 ...
2026-10-12 很多数据分析师能熟练写SQL、做透视表,但当被问到“数据是从哪里来的?经过哪些加工才进入数据仓库?ETL具体做了什么?”时 ...
2026-10-12在传统零售行业同质化竞争激烈、大众营销转化率持续走低的背景下,依托经验与直觉的粗放式营销逐渐失效。美国塔吉特百货(Target ...
2026-10-10随着智能客服、数字化服务、用户精细化运营的快速普及,传统零散、非标准化的客户服务数据与业务体系逐渐难以支撑高效服务治理。 ...
2026-10-10 很多数据分析师能熟练地写SQL、做透视表、算描述性统计,但当被问到“如何预测用户流失概率”“如何归因销量下滑的关键因素 ...
2026-10-10CDA数据分析师 出品 作者:李诗怡 定义区别 · 场景举例 · 指标分类 · 计算方法 知识体系总览 模块 包含内容 一、核 ...
2026-10-09在数据分析工作流中,业务数据大多存储在各类关系型数据库内,例如MySQL、PostgreSQL、SQLite等。Pandas是Python生态中主流的数 ...
2026-10-09在数字化精细化运营时代,海量用户存在需求差异、行为差异、价值差异与偏好差异。如果企业采用“一刀切”的统一营销、统一服务、 ...
2026-10-09 很多数据分析师拿到数据就开始清洗、建模,但当被问到“这批数据属于什么类型——结构化还是非结构化?分类变量还是数值变量 ...
2026-10-09指标体系是企业数字化分析、业务监控、经营决策的核心基础框架,是将零散数据转化为可衡量、可对比、可落地业务价值的关键体系。 ...
2026-10-08随着市场竞争日趋饱和,同质化低价竞争逐渐陷入内卷僵局,传统以价格、渠道、促销为核心的营销模式边际效益持续递减。在此背景下 ...
2026-10-08 很多数据分析师画过趋势图、做过业绩预测,但当被问到“这个月销售额增长20%,到底是长期趋势自然增长,还是促销活动的短期 ...
2026-10-08你有没有想过,手机里点外卖、刷社交软件、转一笔账,背后到底是谁在替你"记着账"? 答案其实很简单:数据库,以及跟它对话的那 ...
2026-10-07CDA数据分析师 出品 作者:李诗怡 一、数据分析四大思维 1. 对比思维:没有对比就没有分析 核心观点:单独一个数字没有意义,有 ...
2026-10-05Kimball 是方法,星型模型是它产出的形状。 很多人把"Kimball vs 星型模型"当成一道选择题——这本身就是个误会:Kimball 是动词 ...
2026-10-05写在开头 老板在微信上甩来一句: "帮我看下为什么销量跌了。" ” 你回工位,打开 SQL,开始写。查订单表、拉近三个月、按 ...
2026-10-03