python - 如何检查 word2vec 训练模型中是否存在 key

Question

我已经使用 Gensim 的文档语料库训练了一个 word2vec 模型。模型训练完成后，我正在编写以下代码来获取单词“view”的原始特征向量。

myModel["view"]

但是，我得到一个单词的 KeyError，这可能是因为它在 word2vec 索引的键列表中不作为键存在。在尝试获取原始特征向量之前，如何检查索引中是否存在键？

Answer 1

Word2Vec 还提供了一个“vocab”成员，您可以直接访问它。

使用pythonistic方法:

if word in w2v_model.vocab:
    # Do something

编辑 自 gensim 2.0 版以来，Word2Vec 的 API 发生了变化。要访问词汇，您现在应该使用以下命令:

if word in w2v_model.wv.vocab:
    # Do something

EDIT 2 属性“wv”已被弃用，将在 gensim 4.0.0 中完全删除。现在回到OP的原始答案:

if word in w2v_model.vocab:
    # Do something