Gensim 4.0.0 得到 model.wv 中的关键字答案

【问题标题】：Gensim 4.0.0 get the word having key in model.wvGensim 4.0.0 得到 model.wv 中的关键字
【发布时间】：2021-10-29 00:00:26
【问题描述】：

with open(args.output, 'w+', encoding='utf-8') as f:
    f.write('%d %d\n' % (len(model.wv), args.size))
    for key in tqdm(model.wv): 
        // code //
        f.write(WORD, VECTOR)

我想创建一个包含 Word 和嵌入向量的 txt 文件。

谁能帮帮我？

【问题讨论】：

标签： python nlp gensim word2vec

【解决方案1】：

不清楚你的意思 - 你能把你的问题编辑得更清楚吗？

在正常的Word2Vec 用法中，wv 对象是KeyedVectors 的一个实例，单词本身就是键。您可以使用单词字符串执行索引访问（Python 括号wv[…]）以获取关联的向量。

如果您想要所有键的列表，您可以：

在model.wv.index_to_key 中查看键列表，其顺序与底层数组中的后备行相同
获取dict 中将键映射到索引位置的所有键：model.wv.key_to_index.keys()

看起来您正在尝试将向量写入简单的文本格式。您可能想改用.save_word2vec_format() 方法：

https://radimrehurek.com/gensim/models/keyedvectors.html#gensim.models.keyedvectors.KeyedVectors.save_word2vec_format

【讨论】：