From 30e4f25155a0e442e5bf0228a36c4a5d2117dec7 Mon Sep 17 00:00:00 2001 From: Timothy Hopper Date: Fri, 7 Aug 2015 14:54:48 -0400 Subject: [PATCH] Fix size of ndz_ in docstring --- lda/lda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lda/lda.py b/lda/lda.py index 7816090..17779d8 100644 --- a/lda/lda.py +++ b/lda/lda.py @@ -45,7 +45,7 @@ class LDA: Alias for `components_` `nzw_` : array, shape = [n_topics, n_features] Matrix of counts recording topic-word assignments in final iteration. - `ndz_` : array, shape = [n_samples, n_features] + `ndz_` : array, shape = [n_samples, n_topics] Matrix of counts recording document-topic assignments in final iteration. `doc_topic_` : array, shape = [n_samples, n_features] Point estimate of the document-topic distributions (Theta in literature)