在 gensim 中计算 tf-idf 的问题

Question

我正在使用 Gensim 为下面提到的我的语料库计算 tf-idf 分数。

corpus=['human interface computer',
 'survey user computer system response time',
 'eps user interface system',
 'system human system eps',
 'user response time']

我目前的代码如下。

dictionary = corpora.Dictionary(line.lower().split() for line in corpus)

class MyCorpus(object):
    def __iter__(self):
        for line in corpus:
            yield dictionary.doc2bow(line.lower().split())

corpus = MyCorpus()

tfidf = models.TfidfModel(corpus)

corpus_tfidf = tfidf[corpus]

但是，我收到错误 RecursionError: maximum recursion depth exceeded while calling a Python object（PS：如果我的代码有误，我很高兴有不同的代码）。请帮我计算我当前语料库的 tf-idf 值。此外，我想在我的语料库中获得 tf-idf 分数最高的 3 个词。

请帮帮我！

Answer 1

好吧，您正在将原始 corpus 列表重新定义为 MyCorpus 对象，所以当尝试对其进行迭代时，这当然会进入无限递归（因为您在 __iter__ 方法中迭代 corpus ）。也许你想定义my_corpus = MyCorpus()？否则请解释您要做什么...