class BaseDriver:
def __init__(self, lda_model=None):
self.lda_model = lda_model
self.me = ModelExporter(lda_model)
def get_all_topics(self, words_in_topic=5):
return self.me.topics(words_in_topic)
"""
t1=Topic(id=1,word_counts=[{'word1':5,'word2':2,'word3':1,'word4':4}])
t2=Topic(id=2,word_counts=[{'word1':5,'word2':2,'word3':1,'word4':4}])
return [t1,t2]
"""
def get_topic_details(self, topic_id, num_words=10):
return self.me.topic(topic_id=topic_id, num_words=num_words)
class BaseDriver:
def __init__(self, lda_model=None):
self.lda_model = lda_model
self.me = ModelExporter(lda_model)
def get_all_topics(self, words_in_topic=5):
return self.me.topics(words_in_topic)
"""
t1=Topic(id=1,word_counts=[{'word1':5,'word2':2,'word3':1,'word4':4}])
t2=Topic(id=2,word_counts=[{'word1':5,'word2':2,'word3':1,'word4':4}])
return [t1,t2]
"""
def get_topic_details(self, topic_id, num_words=10):
return self.me.topic(topic_id=topic_id, num_words=num_words)
def __init__(self, lda_model=None):
self.lda_model = lda_model
self.me = ModelExporter(lda_model)
def __init__(self, lda_model=None):
self.lda_model = lda_model
self.me = ModelExporter(lda_model)
