generate.py 文件源码

def __init__(self, db=None, topic=None, starter=None, enders=['.', '!', '?', ';', ',', ':']):
        self.db = TinyDB(db)
        self.enders = enders
        self.topic = topic
        self.starter = starter
        self.text = None
        self.sentences = []
        self.words = []
        self.bigrams = []
        self.trigrams = []
        self.transitions = defaultdict(list)
        self.starters = []
        self.wordFreq = defaultdict(Counter)
        self.bigramFreq = defaultdict(Counter)

        self.getAll()