使用Gensim获得TF-IDF的单词分数

Question

我试图根据他们的TF-IDF分数在语料库中找到最重要的单词.

按照https://radimrehurek.com/gensim/tut2.html上的示例进行操作.基于

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

TF-IDF得分在每次迭代中都会得到更新.例如,

• Word 0(基于https://radimrehurek.com/gensim/tut1.html的 " 计算机 " )的TF-IDF分数为0.5773(Doc#1),0.4442(Doc#2).
• Word 10(" 图表 ")的TF-IDF分数为0.7071(Doc#7),0.5080(Doc#8),0.4588(Doc#9)

所以这就是我目前如何获得每个单词的最终TF-IDF分数,

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}