本文主要是介绍利用BERT预训练模型的击中返回值,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
利用BERT预训练模型的击中返回值
- get_pooled_output
通过transformer得到的通过全连接简单地获取相应的隐藏状态来“池”模型,返回的到第一个位置,代表该句句子向量【CLS】。
- get_sequence_output
返回transformer最后的结果 - get_all_encoder_layers
返回transformer所有层的结果 - get_embedding_output
返回词向量表示(还没进入transformer)
这篇关于利用BERT预训练模型的击中返回值的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!