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)