def setUp(self):
model = BayesModel()
model.train(u"мама мыла раму", True)
model.train(u"папа пошел гулять", False)
getDefaultDomain().set('testModel', model)
self.message1 = TransitMessage(serialized={ 'text' : u'мама'}).getMessage(getDefaultDomain())
self.message2 = TransitMessage(serialized={ 'text' : u'папа'}).getMessage(getDefaultDomain())
def testTrainClassify(self):
model = BayesModel()
model.train(Texts.pushkin2, True)
model.train(Texts.udaff, False)
return model.classify(Texts.udaff2).addCallback(lambda result: self.assert_(not result)). \
addCallback(lambda _: model.classify(Texts.pushkin)).addCallback(lambda result: self.assert_(result))
def testTrainClassifySimple(self):
model = BayesModel()
model.train(u"мама мыла раму", True)
model.train(u"папа пошел гулять", False)
return model.classify(u"папа").addCallback(lambda result: self.assert_(not result)). \
addCallback(lambda _: model.classify(u"мама")).addCallback(lambda result: self.assert_(result))
def setUp(self):
model = BayesModel()
model.train(u"мама мыла раму", True)
model.train(u"папа пошел гулять", False)
getDefaultDomain().set('testModel', model)