其他分享
首页 > 其他分享> > Bert系列:如何用bert模型输出文本的embedding

Bert系列:如何用bert模型输出文本的embedding

作者:互联网

问题:

分类模型可以输出其文本的embedding吗?LM模型可以输出其文本的embedding吗?答案:可以。

假设你已经用自己的数据fine-tuing好模型。

主要工具设备型号:

python3.6、torch1.7、transformer4.2、macOS、输出huggingface:https://huggingface.co/transformers/model_doc/bert.html

1.用哪个类可以输出文本的embedding?

 

demo:

 

 

标签:Bert,输出,模型,huggingface,embedding,文本,bert
来源: https://blog.csdn.net/pearl8899/article/details/116354207