时间戳

首页 > 解决方案 > 我想为句子中的单个词找到使用 BERT 的词嵌入。有人知道怎么做吗？

python - 我想为句子中的单个词找到使用 BERT 的词嵌入。有人知道怎么做吗？

问题描述

例如在句子中：“地球围绕太阳旋转并围绕它自己的轴旋转”，我想用 BERT 词嵌入来嵌入它，并在这个特定的句子中找到“旋转”的嵌入。怎么做？

标签： pythonnlpbert-language-model

解决方案

天赋使这变得非常容易。首先使用安装库

!pip install flair

然后使用下面的代码，您可以获得句子 AKA 上下文中每个标记的嵌入：

from flair.data import Sentence
s = Sentence('the dummy sentence to be embedded')

from flair.embeddings import BertEmbeddings
bert_embedding = BertEmbeddings()
bert_embedding.embed(s)

for token in s:
   print(token.embedding)

推荐阅读