参考文献:保姆级教程,用PyTorch和BERT进行文本分类 - 知乎 (zhihu.com)
模型地址:https://huggingface.co/bert-base-cased
结果是BaseModelOutput对象:
BaseModelOutput
类型是 Hugging Face Transformers 库中的一个基本输出类型,用于存储预训练模型的输出。该类型通常包含以下属性:
-
last_hidden_state: 这是模型对输入序列的最后一层的隐藏状态。对于BERT模型,这是包含输入序列各个标记的隐藏表示。形状通常是
[batch_size, sequence_length, hidden_size]
,其中batch_size
是输入的批次大小,sequence_length
是输入序列的长度,hidden_size
是隐藏状态的维度。 -
pooler_output: 这是经过池化操作后的表示,通常用于表示整个输入序列的信息。形状通常是
[batch_size, hidden_size]
。 -
hidden_states: 这是模型每一层的隐藏状态的列表。例如,对于BERT-base模型,
hidden_states
包含了所有12层的隐藏状态。形状类似于last_hidden_state
,即[batch_size, sequence_length, hidden_size]
。 -
attentions: 这是模型每一层的注意力权重。形状通常是
[batch_size, num_heads, sequence_length, sequence_length]
。注意,某些模型可能不会在输出中包含注意力权重。
last_hidden_state是一个三维的数据,还需要在进行处理,降低维度
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!