def test_update_from_trained(self):
qd = trainer.Trainer()
qd.train(corpus)
qd2 = trainer.Trainer()
qd2.train(corpus_update)
qd.update_counts_from_trained(qd2)
self.assertEqual(qd._counts, corpus_unigrams)
self.assertEqual(qd._total_docs, corpus_ndocs)
def update(self,query_update,document_update):
# 模型update
qd2 = trainer.Trainer()
qd2.train(query_update)
self.qd.update_counts_from_trained(qd2) # 合并两个容器的训练集
self.query = self.query + query_update
self.document = self.document + document_update
def TrainModel(self):
qd = trainer.Trainer()
qd.train(self.query)
return qd
def _get_qd(self):
qd = trainer.Trainer()
qd.train(corpus)
qd.train(corpus_update)
return qd