实战python局部加权线性回归

利用 python 进行局部加权线性回归实战 ，内含有原始数据及拟合结论图。

# -*- coding: utf-8 -*-
“““
Created on Fri May  4 09:48:38 2018

@author: FZ
“““

import Locally_weighted_linear_regression
import numpy as np
import matplotlib.pyplot as plt

xArryArr = Locally_weighted_linear_regression.loadDataSet（‘ex0.txt‘） #load data

yHat = Locally_weighted_linear_regression.lwlrTest（xArrxArryArr0.003） #prediction


#sort
xMat = np.mat（xArr）
srtInd = xMat[:1].argsort（0）
xSort = xMat[srtInd][:0:]


#drawing

fig = plt.figure（）
ax = fig.add_subplot（111）
ax.plot（xSort[:1]yHat[srtInd]）   #fitting line is different when k changes

ax.scatter（xMat[:1].flatten（）.A[0]np.mat（yArr）.T.flatten（）.A[0]s=2c=‘red‘） #scatter sample point 

#plt.savefig（‘lwlr0_003.png‘dpi=400）
plt.show（）


 

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-05-04 09:59  8.2_locally_weighted_linear_regression\__pycache__\
     文件        1482  2018-05-04 09:59  8.2_locally_weighted_linear_regression\__pycache__\Locally_weighted_linear_regression.cpython-36.pyc
     文件         753  2018-05-04 10:26  8.2_locally_weighted_linear_regression\do_locally_weighted_linear_regression.py
     文件        5600  2011-01-08 10:02  8.2_locally_weighted_linear_regression\ex0.txt
     文件        1638  2018-05-04 09:59  8.2_locally_weighted_linear_regression\Locally_weighted_linear_regression.py
     文件      128139  2018-05-04 10:25  8.2_locally_weighted_linear_regression\lwlr0_003.png
     文件      118787  2018-05-04 10:25  8.2_locally_weighted_linear_regression\lwlr0_01.png
     文件      122805  2018-05-04 10:24  8.2_locally_weighted_linear_regression\lwlr1.png