similarity.py 文件源码

def make_lsi_model(self, seqs):
        if self.use_tfidf:
            seqs = (self.tfidf_model[self.lexicon.doc2bow(tokenize(seq))] for seq in seqs)
        else:
            seqs = (self.lexicon.doc2bow(tokenize(seq)) for seq in seqs)
        self.lsi_model = models.LsiModel(seqs, num_topics=self.n_lsi_dim, id2word=self.lexicon)
        self.lsi_model.save(self.lsi_filepath)
        print "saved lsi model to", self.lsi_filepath