使用Gensim获取TF-国防军单词的分数

Question

我试图在语料库中找到最重要的单词，根据他们的TF-以色列国防军得分。

一直跟随在https://radimrehurek.com/gensim/tut2.html的例子。基于

>>> for doc in corpus_tfidf:
...     print(doc)

TF-以色列国防军的分数在每次迭代中都会被更新.例如,

• 单词0(基于https://radimrehurek.com/gensim/tut1.html的“计算机”)，TF-以色列国防军得分为0.5773 (博士#1)，0.4442 (博士#2)。
• 单词10 (“图”)的TF-以色列国防军得分为0.7071 (博士#7)，0.5080 (博士#8)，0.4588 (博士#9)。

所以我现在得到的是每一个单词的最终TF-以色列国防军得分，

tfidf = gensim.models.tfidfmodel.TfidfModel(corpus)
corpus_tfidf = tfidf[corpus]
d = {}
for doc in corpus_tfidf:
    for id, value in doc:
        word = dictionary.get(id)
        d[word] = value

有更好的办法吗？

提前谢谢。

Answer 1

使用字典理解如何？

d = {dictionary.get(id): value for doc in corpus_tfidf for id, value in doc}