線性迴歸小程序舉例,梯度下降法

用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如下
相關文章
相關標籤/搜索