Ejemplo n.º 1
0
 def __init__(self):
     self.meaningDict={}
     trainList,trainClassifyVec,self.testList,self.testClassifyVec=self.getData()
     vocabList=createVocabList(trainList)
     trainMat=[]
     for postingData in trainList:
         trainMat.append(bagOfWords2VecMN(vocabList,postingData))
     self.p0V,self.p1V,self.pAb=trainNB(trainMat,trainClassifyVec)
     self.vocabList=vocabList
Ejemplo n.º 2
0
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)