main.py 文件源码

def costFunc(alpha, *args):
    i = args[2]
    original_thetai = args[0]
    delta_thetai = args[1]
    x = args[3]
    y = args[4]
    _lambda = args[5]
    labels = set(y)
    thetai = original_thetai
    thetai[i, :] = thetai[i, :] - alpha * delta_thetai
    k = 0
    sum_log_p = 0.0
    for label in labels:
        index = y == label
        xi = x[index]
        p = condProb(original_thetai,thetai[k, :], xi)
        log_p = np.log10(p)
        sum_log_p = sum_log_p + log_p.sum()
        k = k + 1
    r = -sum_log_p / x.shape[0]+ (_lambda / 2.0) * pow(norm(thetai),2)
    #print r ,alpha
    return r