Gensim 保存的字典没有 id2token

Question

我已将 Gensim 字典保存到磁盘。当我加载它时，id2token 属性字典没有被填充。

保存字典的一段简单代码：

dictionary = corpora.Dictionary(tag_docs)
dictionary.save("tag_dictionary_lda.pkl")

现在当我加载它时（我正在将它加载到一个 jupyter 笔记本中），它仍然可以很好地将令牌映射到 ID，但 id2token 不起作用（我无法将 ID 映射到令牌）事实上 @987654324 @ 根本没有填充。

> dictionary = corpora.Dictionary.load("../data/tag_dictionary_lda.pkl")
> dictionary.token2id["love"]
Out: 1613

> dictionary.doc2bow(["love"])
Out: [(1613, 1)]

> dictionary.id2token[1613]
Out: 
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input> in <module>()
----> 1 dictionary.id2token[1613]

KeyError: 1613

> list(dictionary.id2token.keys())
Out: []

有什么想法吗？

Answer 1

您不需要dictionary.id2token[1613]，因为您可以直接使用dictionary[1613]。

请注意，如果您之后检查dictionary.id2token，它将不再为空。这是因为dictionary.id2token 仅在请求节省内存时才形成（正如在 Dictionary 类的初始化期间所述）。