Fix Bug: check parameter topic for top_terms_of_topic and ADD: print Doc-Topic Matrix and Topic-Term Matrix in example.py

Author: JoeZJH

README.md

@@ -115,6 +115,8 @@ llda_model_new.load_model_from_dir(save_model_dir, load_derivative_properties=Fa
 print "llda_model_new", llda_model_new
 print "llda_model", llda_model
 print "Top-5 terms of topic 'negative': ", llda_model.top_terms_of_topic("negative", 5, False)
+print "Doc-Topic Matrix: \n", llda_model.theta
+print "Topic-Term Matrix: \n", llda_model.beta
 ```
 
 

example/example.py

@@ -72,3 +72,5 @@
 print "llda_model_new", llda_model_new
 print "llda_model", llda_model
 print "Top-5 terms of topic 'negative': ", llda_model.top_terms_of_topic("negative", 5, False)
+print "Doc-Topic Matrix: \n", llda_model.theta
+print "Topic-Term Matrix: \n", llda_model.beta

model/labeled_lda.py

@@ -938,6 +938,8 @@ def top_terms_of_topic(self, topic, k, with_probabilities=True):
  :param k: int, the number of terms
  :return: the top-k terms of topic
  """
+ if topic not in self.topic_vocabulary:
+ raise Exception("Cannot find topic \"%s\"" % topic)
  beta = self.beta_k(self.topic_vocabulary[topic])
  terms = sorted(list(zip(self.terms, beta)), key=lambda x: x[1], reverse=True)
  if with_probabilities: