from lda_for_fragments import Ms2Lda
ms2lda = Ms2Lda.resume_from('notebooks/results/beer3pos.project')
ms2lda.do_thresholding(th_doc_topic=0.05, th_topic_word=0.01)
ms2lda.print_topic_words()
special_nodes = [
('doc_21213', '#CC0000'), # maroon
('doc_21758', 'gold'),
('doc_21182', 'green'),
('topic_240', '#CC0000'), # maroon
('topic_76', 'aqua'),
('topic_253', '#ff1493') # deep pink
]
ms2lda.plot_lda_fragments(consistency=0.0, interactive=True, to_highlight=special_nodes)
