朴素贝叶斯分类器训练函数:
def trainNB0(trainMatrix, trainCategory):
numTrainDocs = len(trainMatrix) #文档个数
numWords = len(trainMatrix[0]) #一个文档中的单词数
pAbusive = sum(trainCategory)/float(numTrainDocs) #计算文档为侮辱性文档的概率
p0Num = zeros(numWords)
p1Num = zeros(numWords)
p0Denom = 2.0; p1Denom = 2.0
for i in range(numTrainDocs):
if trainCategory[i] == 1:
p1Num += trainMatrix[i]
p