Fix size of ndz_ in docstring

ariddell · Aug 7, 2015 · 30e4f25 · 30e4f25
1 parent 9a536e0
commit 30e4f25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)