首页 > 代码库 > 机器学习入门——单变量线性回归
机器学习入门——单变量线性回归
线性回归的概念,在高中数学书里就出现过。
给你一些样本点,如何找出一条直线,使得最逼近这些样本点。
给出一个例子:假设 x 是房子面积,y是房子价格,确定一条直线需要theta0和theta1.
给出x,我们就可以计算出房子的价格 h(x) = theta0+theta1*x
关键是如何计算出theta0和theta1,也就是如何找出这么一条直线呢?
在这里,引入一个概念,叫做cost function.m表示样本个数,也就是训练样本数目
这是一个square error,学过统计的应该经常见到
因此,我们的目的i就变成如何最小化这个J,意味着这条直线最逼近我们的样本点
先简化一下问题,假设theta0 = 0,那么我们的目标就是最小化J(theta1)
假设目前有3个样本点(1,1)(2,2) (3,3)
J(0) = 1/(2*3)*((1-0)^2+(2-0)^2+(3-0)^2)
当theta1取不同的值时,J(theta1)就形成了一个二次函数,当theta1 = 1是极小值也是最小值。
问题回到2元函数theta0,theta1
容易想像,这肯定是一个平面函数了
关键问题就是如何找到这个最低点
下面再举个例子直观的感受下,不再使用3维图了,而是使用如下右图。
左图的直线明显和样本点一点都不逼近,所以在右图中的红叉就离中间比较远
这里比上面稍微好点,于是红点离中心更近了
这是最优结果,落在了中间
还是那个问题,如何寻找我们的theta0和theta1呢?我们可以用梯度下降的方法。如图:
随机初始化theta0和theta1,一直往梯度下降的方向走,J就会越来越小。公式如下:
其中,alpha是我们的learning rate,不能太小,否则算法速度会很慢,也不能太大,否则很容易越过最小值导致不能收敛。
我们在前面有:
于是
因此,我们的算法变成
当你发现两次循环之间的theta0和theta1的差别很小很小就converge了,你可以设置一个阀值比如10e-6。
配套练习在http://download.csdn.net/detail/abcd1992719g/7306053