贝叶斯估计算法过程

贝叶斯估计是最大似然估计的改进。

在最大似然估计算法中计算得到的先验概率、条件概率都有可能是0,这会导致最终得到的后验概率没有意义。 贝叶斯估计在最大似然估计的分子分母中同时添加了系数lamda,防止出现计算结果为0的情况。

def NaiveteBayes(T, y, a, Y, x, lam=0):
    # 计算先验概率
    prepro = {}
    for yRange in Y:
        #print (yRange, Y[Y==yRange].shape[0], )
        prepro[yRange] = (y[y==yRange].shape[0]+lam)/(y.shape[0]+lam*Y.shape[0])
    print('先验概率:',prepro)
    # 计算条件概率
    conpro = {}
    for i in range(len(a)):  # 遍历每个特征
        for j in a[i]: # 遍历特征的每个取值
            for k in Y:
                numerator = X[(y==k)&(X[:,i]==j),:].shape[0]+lam
                denominator = X[y==k,:].shape[0]+lam*a[i].shape[0]
                conpro[(i,j, k)] = numerator/denominator
                # print (i,j,k,numerator, denominator)
    print('条件概率:',conpro)
    # 计算后验概率的分子
    postpro = {}
    for yRange in Y:
        pro = 1
        for i in range(x.shape[0]):
            pro = pro * conpro[(i, x[i], yRange)]
            print (conpro[(i, x[i], yRange)])
        postpro[yRange] = pro * prepro[yRange]
    print ('后验概率', postpro)
    # 确定X的分类
    import operator
    return sorted(postpro.items(),   # iterable -- 可迭代对象,在python2中使用A.iteritems(),在python3中使用A.items()
           key=operator.itemgetter(1),   # key -- 主要是用来进行比较的元素,指定可迭代对象中的一个元素来进行排序,这里指基于item的value进行排序
           reverse=True)[0][0]   # reverse -- 排序规则,reverse = True 降序 , reverse = False 升序(默认)。
# 排序结果是一个list

Last updated