机器学习

基于概率论的分类方法:朴素贝叶斯

2019-03-12  本文已影响0人  CSTDOG

基于贝叶斯决策理论的分类方法

使用朴素贝叶斯进行文档分类

p(c_i|x,y)=p(x,y|c_i)p(c_i)/p(x,y)
graph TD
收集数据/可以使用任何方法-->准备数据/需要数值型或者布尔型数据
准备数据/需要数值型或者布尔型数据-->分析数据/有大量特征时,绘制特征作用不大,此时使用直方图效果更好
分析数据/有大量特征时,绘制特征作用不大,此时使用直方图效果更好-->训练算法/计算不同的独立特征的条件概率
训练算法/计算不同的独立特征的条件概率-->测试算法/计算错误率
测试算法/计算错误率-->使用算法
p(w_0|c_i)....P(w_n|c_i)=P(w_0,....,w_n|c_i)

文本分类的代码实现

#词表到向量的转换函数
def createVocbList(dataSet):
    vocabSet=set([])
    for document in dataSet:
        vocabSet= vocabSet | set(document)
    return vocabSet
#文档到文本向量
def setOfWords2Vec(vocabList, inputSet):
    returnVec = [0]*len(vocabList)
    for word in inputSet:
        if word in vacabList:
            returnVec[vocabList.index(word)]=1
        else:
            print("the word:" + word +"is no in our Vocabulary")
    return returnVec
#训练算法:从词向量计算概率
#输入:trainMatrix-文档矩阵,trainCategory-文档标签所构成的向量
def trainNbB0(trainMatrix, trainCategory):
    numTrainDocs = len(trainMatrix)
    numWord = len(trainMatrix[0])
    #计算P(ci),由于这里是处理二分类问题,因此只需要计算P(c1),即侮辱性语句即可
    pAbusive = sum(trainCategory)/float(numTrainDocs)
    #计算各个词在不同类别中的总数
    p0Num = zeros(numWord)
    p1Num = zeros(numWord)
    #计算不同类别的总次数
    p0Demon = 0.0
    p1Denom = 0.0
    for i in range(numTrainDocs):
        if trainCategory[i] == 1:
            p0Num += trainMatrix[i]
            p0Demon += sum(trainMatrix[i])
        else:
            p1Num += trainMatrix[i]
            p1Demon += sum(trainMatrix[i])
    p0Vect = p0Num/p0Demon
    p1Vect = p1Num/p1Denom
    return pAbusive,p0Vect,p1Vect
#朴素贝叶斯词袋模型
def bagOfWords2VecMN(vocabList, inputSet):
    returnVec =[0]*len(vocabList)
    for word in inputSet:
        if word in vocabList:
            returnVec[vocabList.index(word)] +=1
    return returnVec
上一篇 下一篇

猜你喜欢

热点阅读