二元组到向量

Question

我想使用 word2vec 工具为文档构建词嵌入。我知道如何找到对应于单个单词（unigram）的向量嵌入。现在，我想找到一个二元组的向量。可以使用 word2vec 吗？如果是，如何？

Answer 1

以下代码片段将为您提供二元语法的向量表示。请注意，要转换为向量的二元组需要在单词之间使用下划线而不是 space，例如bigram2vec(unigrams, "this report")错了，应该是bigram2vec(unigrams, "this_report")。有关生成 unigram 的更多详细信息，请参阅 gensim.models.word2vec.Word2Vec class here.

from gensim.models import word2vec

def bigram2vec(unigrams, bigram_to_search):
    bigrams = Phrases(unigrams)
    model = word2vec.Word2Vec(bigrams[unigrams])
    if bigram_to_search in model.vocab.keys():
        return model[bigram_to_search]
    else:
        return None

二元组到向量

Bigram to a vector

nlp

word2vec

word-embedding