Source code for topik.models.tests.test_lda
import nose.tools as nt
from topik.models.tests.test_data import test_vectorized_output
from topik.models.lda import lda
[docs]def test_train():
model_output = lda(test_vectorized_output, ntopics=2)
for doc in model_output.doc_topic_matrix.values():
nt.assert_almost_equal(sum(doc), 1)
for topic in model_output.topic_term_matrix.values():
nt.assert_almost_equal(sum(topic), 1)