Commit 19f7f01a authored by Kathryn Elliott's avatar Kathryn Elliott

Minimum topic count & exporting information

* Changed min topic count to 5
* Added export topics per document
* Added export topic visualisations (ie pyLDAvis
coherency model visualisations)
parent 57d7224e
......@@ -51,7 +51,7 @@ if (len(sys.argv) < 2):
file = sys.argv[1]
topic_model = TopicModel(load_data(file), min_topic_count=8, max_topic_count=20, extra_stopwords=extra_stopwords, extra_non_stopwords=extra_non_stopwords)
topic_model = TopicModel(load_data(file), min_topic_count=5, max_topic_count=20, extra_stopwords=extra_stopwords, extra_non_stopwords=extra_non_stopwords)
topic_model.pre_process_corpus()
topic_model.run_lda()
......@@ -66,3 +66,8 @@ print("################ " + str(n) + " ################")
print("########################################")
topic_model.print_topics(n)
print("")
print("")
topic_model.export_topics_per_documents(n)
topic_model.export_topic_visualisation_data(n)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment