pyLDAvis:尝试可视化主题时发生验证错误

Hac*_*rds 6 python nlp lda topic-modeling

我尝试使用gensim生成300000条记录的主题。在尝试可视化主题时,出现验证错误。我可以在模型训练后打印主题,但是在使用pyLDAvis时失败

# Running and Training LDA model on the document term matrix.
ldamodel1 = Lda(doc_term_matrix1, num_topics=10, id2word = dictionary1, passes=50, workers = 4)

(ldamodel1.print_topics(num_topics=10, num_words = 10))
 #pyLDAvis
d = gensim.corpora.Dictionary.load('dictionary1.dict')
c = gensim.corpora.MmCorpus('corpus.mm')
lda = gensim.models.LdaModel.load('topic.model')

#error on executing this line
data = pyLDAvis.gensim.prepare(lda, c, d)
Run Code Online (Sandbox Code Playgroud)

在pyLDAvis上面运行后,我尝试执行以下错误

---------------------------------------------------------------------------
ValidationError                           Traceback (most recent call last)
<ipython-input-53-33fd88b65056> in <module>()
----> 1 data = pyLDAvis.gensim.prepare(lda, c, d)
      2 data

C:\ProgramData\Anaconda3\lib\site-packages\pyLDAvis\gensim.py in prepare(topic_model, corpus, dictionary, doc_topic_dist, **kwargs)
    110     """
    111     opts = fp.merge(_extract_data(topic_model, corpus, dictionary, doc_topic_dist), kwargs)
--> 112     return vis_prepare(**opts)

C:\ProgramData\Anaconda3\lib\site-packages\pyLDAvis\_prepare.py in prepare(topic_term_dists, doc_topic_dists, doc_lengths, vocab, term_frequency, R, lambda_step, mds, n_jobs, plot_opts, sort_topics)
    372    doc_lengths      = _series_with_name(doc_lengths, 'doc_length')
    373    vocab            = _series_with_name(vocab, 'vocab')
--> 374    _input_validate(topic_term_dists, doc_topic_dists, doc_lengths, vocab, term_frequency)
    375    R = min(R, len(vocab))
    376 

C:\ProgramData\Anaconda3\lib\site-packages\pyLDAvis\_prepare.py in _input_validate(*args)
     63    res = _input_check(*args)
     64    if res:
---> 65       raise ValidationError('\n' + '\n'.join([' * ' + s for s in res]))
     66 
     67 

ValidationError: 
 * Not all rows (distributions) in topic_term_dists sum to 1.
Run Code Online (Sandbox Code Playgroud)

Azu*_*reX 5

发生这种情况是因为 pyLDAvis 程序期望模型中的所有文档主题至少在语料库中出现一次。当您在制作语料库/文本之后和制作模型之前进行一些预处理时,可能会发生这种情况。

模型内部词典中未在您提供的词典中使用的单词将导致此操作失败,因为现在概率略小于 1。

您可以通过将缺少的单词添加到语料库字典中(或将单词添加到语料库并从中创建字典)来解决此问题,或者您可以将此行添加到 site-packages\pyLDAvis\gensim.py 代码中的“assert”之前topic_term_dists.shape[0] == doc_topic_dists.shape[1]" (应该是〜第 67 行)

topic_term_dists = topic_term_dists / topic_term_dists.sum(axis=1)[:, None]
Run Code Online (Sandbox Code Playgroud)

假设您的代码运行到那时,这应该重新规范化主题分布,而不会丢失字典项。但请注意,最好将所有术语都包含在语料库中。