京公网安备 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
教材入口:https://edu.cda.cn/goods/show/3151 “纲举目张,执本末从。” 若想在数据分析领域有所收获,一套合适的学习教材至 ...
2025-11-04【2025最新版】CDA考试教材:CDA教材一级:商业数据分析(2025)__商业数据分析_cda教材_考试教材 (cdaglobal.com) ...
2025-11-04在数字化时代,数据挖掘不再是实验室里的技术探索,而是驱动商业决策的核心能力 —— 它能从海量数据中挖掘出 “降低成本、提升 ...
2025-11-04在 DDPM(Denoising Diffusion Probabilistic Models)训练过程中,开发者最常困惑的问题莫过于:“我的模型 loss 降到多少才算 ...
2025-11-04在 CDA(Certified Data Analyst)数据分析师的工作中,“无监督样本分组” 是高频需求 —— 例如 “将用户按行为特征分为高价值 ...
2025-11-04当沃尔玛数据分析师首次发现 “啤酒与尿布” 的高频共现规律时,他们揭开了数据挖掘最迷人的面纱 —— 那些隐藏在消费行为背后 ...
2025-11-03这个问题精准切中了配对样本统计检验的核心差异点,理解二者区别是避免统计方法误用的关键。核心结论是:stats.ttest_rel(配对 ...
2025-11-03在 CDA(Certified Data Analyst)数据分析师的工作中,“高维数据的潜在规律挖掘” 是进阶需求 —— 例如用户行为包含 “浏览次 ...
2025-11-03在 MySQL 数据查询中,“按顺序计数” 是高频需求 —— 例如 “统计近 7 天每日订单量”“按用户 ID 顺序展示消费记录”“按产品 ...
2025-10-31在数据分析中,“累计百分比” 是衡量 “部分与整体关系” 的核心指标 —— 它通过 “逐步累加的占比”,直观呈现数据的分布特征 ...
2025-10-31在 CDA(Certified Data Analyst)数据分析师的工作中,“二分类预测” 是高频需求 —— 例如 “预测用户是否会流失”“判断客户 ...
2025-10-31在 MySQL 实际应用中,“频繁写入同一表” 是常见场景 —— 如实时日志存储(用户操作日志、系统运行日志)、高频交易记录(支付 ...
2025-10-30为帮助教育工作者、研究者科学分析 “班级规模” 与 “平均成绩” 的关联关系,我将从相关系数的核心定义与类型切入,详解 “数 ...
2025-10-30对 CDA(Certified Data Analyst)数据分析师而言,“相关系数” 不是简单的数字计算,而是 “从业务问题出发,量化变量间关联强 ...
2025-10-30在构建前向神经网络(Feedforward Neural Network,简称 FNN)时,“隐藏层数目设多少?每个隐藏层该放多少个神经元?” 是每个 ...
2025-10-29这个问题切中了 Excel 用户的常见困惑 —— 将 “数据可视化工具” 与 “数据挖掘算法” 的功能边界混淆。核心结论是:Excel 透 ...
2025-10-29在 CDA(Certified Data Analyst)数据分析师的工作中,“多组数据差异验证” 是高频需求 —— 例如 “3 家门店的销售额是否有显 ...
2025-10-29在数据分析中,“正态分布” 是许多统计方法(如 t 检验、方差分析、线性回归)的核心假设 —— 数据符合正态分布时,统计检验的 ...
2025-10-28箱线图(Box Plot)作为展示数据分布的核心统计图表,能直观呈现数据的中位数、四分位数、离散程度与异常值,是质量控制、实验分 ...
2025-10-28在 CDA(Certified Data Analyst)数据分析师的工作中,“分类变量关联分析” 是高频需求 —— 例如 “用户性别是否影响支付方式 ...
2025-10-28