Извлечь матрицу вероятностей словесных слов в gensim LdaModel

Question

У меня есть модель LDA и вероятности документа.

# build the model on the corpus 
ldam = LdaModel(corpus=corpus, num_topics=20, id2word=dictionary) 
# get the document-topic probabilities 
theta, _ = ldam.inference(corpus) 

Мне также необходимо распределить слова для всех тем, то есть матрицы вероятностей по тематическому слову. Есть ли способ извлечь эту информацию?

Спасибо!

Answer 1

Тема перспектива матрица (лямбда) доступна через:

topics_terms = ldam.state.get_lambda() 

Если вы хотите распределение вероятностей просто нормализовать его:

topics_terms_proba = np.apply_along_axis(lambda x: x/x.sum(),1,topics_terms)