在gensim word2vec中随机选择向量

Question

我使用 gensim 训练了一个 word2vec 模型，我想从中随机选择向量，并找到相应的单词。 最好的方法是什么？

Answer 1

如果您的Word2Vec模型实例在变量model ，那么在model.wv.index2word有模型已知的所有单词的列表。 （旧版本的 gensim 中的属性略有不同。）

因此，您可以使用 Python 内置的choice()方法在random模块中choice()一项：

import random
print(random.choice(model.wv.index2entity) 

Answer 2

如果您想使用 Gensim 4.0.0 从 word2vec 中获取n 个随机单词（键），只需使用random.sample ：

import random
import gensim
# Here we use Gensim 4.0.0
w2v = gensim.models.KeyedVectors.load_word2vec_format("model.300d")
# Get 10 random words (keys) from word2vec model
random_words = random.sample(w2v.index_to_key, 10)
print("Random words: "+ str(random_words))

一块蛋糕:)