线性回归 
用线性回归找到最佳拟合直线
回归的目的是预测数值型数据,根据输入写出一个目标值的计算公式,这个公式就是回归方程(regression equation),变量前的系数(比如一元一次方程)称为回归系数(regression weights)。求这些回归系数的过程就是回归。
假设输入数据存放在矩阵
from numpy import *
def loadDataSet(fileName):
    numFeat = len(open(fileName).readline().split(‘\t‘))-1
    dataMat = []; labelMat = []
    fr = open(fileName)
    for line in fr.readlines():
        lineArr = []
        curLine = line.strip().split(‘\t‘)
        for i in range(numFeat):
            lineArr.append(float(curLine[i]))
        dataMat.append(lineArr)
        labelMat.append(float(curLine[-1]))
    return dataMat, labelMat这部分数据是由tap分隔,并且最后一个值是目标值。接下来计算回归系数:
def standRegres(xArr, yArr):
    xMat = mat(xArr); yMat = mat(yArr).T
    xTx = xMat.T*xMat
    if linalg.det(xTx) == 0.0:
        print "This matrix is singular, cannot do inverse"
        return
    ws = xTx.I * (xMat.T*yMat)
    return ws该函数首先读入linalg.det() 来计算行列式。好了,我们来看看效果,首先读入数据:
 xArr,yArr=loadDataSet(‘ex0.txt‘)我们先看看前两条数据:
print xArr[0:2]
[[1.0, 0.067732], [1.0, 0.42781]]第一个值总是等于1.0,也就是
现在看看计算回归系数函数效果:
 ws = standRegres(xArr,yArr)
 print ws
[[ 3.00774324]
 [ 1.69532264]]现在得到了回归系数,那么我们可以通过回归方程进行预测yHat:
xMat = mat(xArr)
yMat = mat(yArr)
yHat = xMat*ws为方便观察,我们画出数据集散点图:
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(xMat[:,1].flatten().A[0], yMat.T[:,0].flatten().A[0])现在绘制最佳拟合直线,那么需要画出预测值yHat,如果直线上数据点次序混乱,绘图时将会出现问题,所以要将点按照升序排序:
xCopy = xMat.copy()
xCopy.sort(0)
yHat=xCopy*ws
ax.plot(xCopy[:,1], yHat)
plt.show()我们来看看效果: 
 
最佳拟合直线方法将数据视为直线进行建模,但图中的数据集似乎还有更好的拟合方式。
局部加权线性回归
线性回归可能出现欠拟合的现象,如上图所示,因为它求的是具有最小均方误差的无偏差估计。所以有些方法允许在估计中引入一些偏差,从而降低预测的均方误差。其中一个方法就是使用局部加权线性回归(Locally Weighted Linear Regression,LWLR),我们给待测点附近的每个点赋予一定的权重,是用此方法解出的回归系数如下:
LWLR使用“核”来对附近的点赋予更高的权重,核的类型可以自由选择,最常用的核就是高斯核,高斯核对应的权重如下:
def lwlr(testPoint, xArr, yArr, k = 1.0):
    xMat = mat(xArr); yMat = mat(yArr).T
    m = shape(xMat)[0]
#    eye()返回一个对角线元素为1,其他元素为0的二维数组。 
    weights = mat(eye((m)))
    for j in range(m):
        diffMat = testPoint - xMat[j,:]
        weights[j,j] = exp(diffMat*diffMat.T/(-2.0*k**2))
    xTx = xMat.T * (weights * xMat)
    if linalg.det(xTx) == 0:
        print "This matrix is singular, cannot do inverse"
        return
    ws = xTx.I * (xMat.T * (weights * yMat))
    return testPoint * ws
def lwlrTest(testArr, xArr, yArr, k=1.0):
    m = shape(testArr)[0]
    yHat = zeros(m)
    for i in range(m):
        yHat[i] = lwlr(testArr[i],xArr,yArr,k)
    return yHatlwlr()函数得到的是单点的估计,为得到所有点的估计,可以调用lwlrTest()函数。
yHat = lwlrTest(xArr,xArr,yArr,1.0)下面绘出估计值和原始值,看看yHat的拟合效果。需要将数据点按序排列。
xMat = mat(xArr)
srtInd = xMat[:,1].argsort(0)
xSort = xMat[srtInd][:,0,:]然后使用Matplotlib绘图:
 fig = plt.figure()
 ax = fig.add_subplot(111)
 ax.plot(xSort[:,1],yHat[srtInd])
 ax.scatter(xMat[:,1].flatten().A[0], mat(yArr).T.flatten().A[0], s = 2, c = ‘red‘)
 plt.show()当k=1.0时,权重很大,相当于将所有数据视为等权重,得出的最佳拟合直线与标准线性回归一致: 
 
当使用k=0.01时:
yHat = lwlrTest(xArr,xArr,yArr,0.01)结果如下: 
 
当使用k=0.003时:
yHat = lwlrTest(xArr,xArr,yArr,0.003)结果如下: 
 
由上三种结果可以看出,在k=1.0时,得出的拟合曲线欠拟合,k=0.01时,曲线拟合效果不错,而k=0.003时,又有些过拟合。
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文地址:http://blog.csdn.net/lxytsos/article/details/49225451