用GloVe将单词转换成矢量

Question

我把我的手套包装了如下：

import gensim.downloader as api
model = api.load("glove-wiki-gigaword-100")

并希望创建一个函数，在其中传递一个word和GloVe模型，它将返回相应的向量，例如，

def convert_word_to_vec(word, model):

当我传入convert_word_to_vec(lol, model)时，它将返回单词lol的向量

有办法绕道吗？谢谢!

Answer 1

用法：

import gensim.downloader as api
model = api.load("glove-wiki-gigaword-100")

vector = model['lol']
print(vector)  # array with shape (100,)

请查看文档以获得更多选项：

https://radimrehurek.com/gensim/models/keyedvectors.html