Пример #1
0
    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())
Пример #2
0
 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))
Пример #3
0
 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))
Пример #4
0
 def setUp(self):
     model = BayesModel()
     model.train(u"мама мыла раму", True)
     model.train(u"папа пошел гулять", False)
     getDefaultDomain().set('testModel', model)