def multiTest(self):
xArr, yArr = self.loadDataSet('data/reg/ex0.txt')
reg = Regression()
yHat = reg.lwlrTest(xArr, xArr, yArr, 0.01)
xMat = mat(xArr)
strInd = xMat[:, 1].argsort(0)
xSort = xMat[strInd][:, 0, :]
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(xMat[:, 1].flatten().A[0], mat(yArr).T[:, 0].flatten().A[0])
ax.plot(array(xSort[:, 1]), array(yHat[strInd]))
plt.show()
def simpleTest(self):
xArr, yArr = self.loadDataSet('data/reg/ex0.txt')
print xArr[0:2]
reg = Regression()
ws = reg.standRegres(xArr, yArr)
print ws
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])
xCopy = xMat.copy()
xCopy.sort(0)
yHat = xCopy * ws
m = xCopy[:, 1]
ax.plot(array(m), array(yHat))
plt.show()
