Logistic回归二分类问题实现

首先介绍一下Logistic函数

y=\frac{1}{1+e^{-z}}

形状呢是一个S形

那么对一个二分类问题y=\{0,1\},考虑线性回归函数z=w^Tx+b,我们可以用Logistic函数将z转换为0, 1值,这样就得到了Logistic回归模型

y=\frac{1}{1+e^{-(w^Tx+b)}}

那么接下来需要确定式中的w和b。通过极大似然法来估计w和b,给定数据集D=\{(x_1,y_1),...,(x_m,y_m)\},Logistic回归模型最大化对数似然(log-likehood)

l(w,b)=\sum_{i=1}^{m}lnp(y_i|x_i;w,b)

其中的似然项

p(y_i|x_i;w,b)=y_ip_1(\hat{x_i};w,b)+(1-y_i)p_0(\hat{x_i};w,b)

将似然项代入对数似然,并且最大化对数似然,相当于最大化

l(w;b)=\sum_{i=1}^{m}((y_i-1)(w^Tx+b)-ln(1+e^{-(w^Tx+b)}))

这是一个凹函数,加个负号可以变成凸函数,就可以根据凸优化理论运用牛顿法来求最优解。在Logistic回归中,求w的迭代公式

w^{t+1}\leftarrow w^t-H^{-1}\nabla _{w^t}(l(w^t))

其中\nabla_w是梯度算子,H()是Hessian矩阵

\nabla_w(l(w))=\sum_{i=1}^{m}(y_i-u(x_i))x_i=X^T(y-u)

H=\nabla_w\nabla_w(l(w^t))=\sum_{i=1}^{m}u(x_i)(u(x_i)-1)x_ix_i^T=X^TRX

带到迭代公式里面

w^{t+1}\leftarrow w^t-(X^TRX)^{-1}X^T(u-y)

那么下面上代码。参考了《机器学习实战》,用的数据集是网上随便找的,一共三列,两个属性一个label,共100条记录

from numpy import *
import matplotlib.pyplot as plt

# 读取数据
def loadDataSet():
    dataMat = []
    labelMat = []
    fr = open('D:/banana/test.txt')
    for line in fr.readlines():
        lineArr = line.strip().split()
        dataMat.append([1.0, float(lineArr[0]), float(lineArr[1])])
        labelMat.append(int(float(lineArr[2])))
    return dataMat, labelMat

# sigmoid函数
def sigmoid(inX):
    return 1.0 / (1 + exp(-inX))

# 随机梯度上升优化算法
def stocGradAscent(dataMatrix, classLabels, numIter=150):
    m, n = shape(dataMatrix)
    weights = ones(n)
    for j in range(numIter):
        dataIndex = list(range(m))
        for i in range(m):
            alpha = 4 / (1.0 + j + i) + 0.01  # 每次迭代时需调整
            randIndex = int(random.uniform(0, len(dataIndex)))  # 随机选取更新
            h = sigmoid(sum(dataMatrix[randIndex] * weights))
            error = classLabels[randIndex] - h
            weights = weights + alpha * dot(error, dataMatrix[randIndex])
            del(dataIndex[randIndex])
    return weights

# 画出决策边界
def plotBestFit(weights):
    dataMat, labelMat = loadDataSet()
    dataArr = array(dataMat)
    n = shape(dataArr)[0]
    xcord1 = []
    ycord1 = []
    xcord2 = []
    ycord2 = []
    for i in range(n):  # 分类
        if int(labelMat[i]) == 1:
            xcord1.append(dataArr[i, 1])
            ycord1.append(dataArr[i, 2])
        else:
            xcord2.append(dataArr[i, 1])
            ycord2.append(dataArr[i, 2])
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.scatter(xcord1, ycord1, s=30, c='red', marker='s')
    ax.scatter(xcord2, ycord2, s=30, c='green')
    x = arange(-4.0, 4.0, 0.1)
    y = (-weights[0] - weights[1] * x) / weights[2]  # 最佳拟合直线
    ax.plot(x, y)
    plt.xlabel('X1')
    plt.ylabel('X2')
    plt.show()

运行结果是这样

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值