def classifyNB(self,listWord2classify): ''' 对给定的词组列表进行分类 :param vec2classify: :return: ''' vectorWord2classify=bagOfWords2VecMN(self.vocabList,listWord2classify) res=classifyNB(vectorWord2classify,self.p0V,self.p1V,self.pAb) return res
def testTrainNB(): listOPosts,trainCategory=loadDataSet() vocabList=createVocabList(listOPosts) trainMat=[] for postinDoc in listOPosts: trainMat.append(setOfWords2Vec(vocabList,postinDoc)) p0V,p1V,pAb=trainNB(trainMat,trainCategory) testEntry=['love','wang','xi','my','dalmation'] vec2classify=array(setOfWords2Vec(vocabList,testEntry)) print testEntry,'classify as:',classifyNB(vec2classify,p0V,p1V,pAb)