线性回归小程序举例,梯度下降法

用matlab写了线性回归小程序,主程序如下:后面调用了函数,名字function1
通过一些坐标点,来线性回归拟合一条直线y=ax+b,学习步长theta可以设置,一般0.01,0.001。。。。
function1里用的梯度下降法,
a=a-theta*a的偏导数
b=b-theta*b的偏导数
返回m、n是新的a和b,tt是最小二乘法的方差结果,可以设定小于一定的值就停止


test.m如下


coex=[1 2 3 4 5 6 7 8 9 10];
coey=[2 4 5 7 9 11 13.5 16.2 18.2 20.2];


a=2;
b=2;
theta=0.01;


for idx=1:100000
    [m,n,tt] = function1(coex,coey,a,b,theta);
    a=m;
    b=n;
    fprintf('%d:  a:%f,  b:%f,  tt:%f\n',idx,a,b,tt);
    if tt<0.2
        break
    end
end


plot(coex,coey,'*r');
hold on;


yy=a*coex+b;
plot(coex,yy,'-b');
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值