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:

 

 

上一篇:git命令简写配置


下一篇:huggingface tokenizer一次性转换和分布转换