局部加权回归(LWR) Matlab模板

将百度文库上一份局部加权回归的代码,将其改为模板以便复用。

q2x,q2y为数据集,是n*1的矩阵;

r是波长参数,就是对于距离的惩罚力度;

q_x是要拟合的数据横坐标,是1*n的矩阵;

得到的q_y即为所求坐标,是n*1的矩阵。

由于使用的是正规方程而非最小化代价函数来求方程,这份代码就留给建模的时候使用吧,毕竟对于多变量或特殊情况,正规方程效果并不好。

function q_y = LWR(q2x, q2y, r, q_x)
q2x=[ones(size(q2x,1),1) q2x];
m = size(q2x,1);

figure;
hold on;
plot(q2x(:,2),q2y,'.r');

q_y=zeros(size(q_x,2),1);

for k=1:1:size(q_x,2)
    w=zeros(m,m);
    for i=1:1:m        
        w(i,i)=exp(-((q_x(k)-q2x(i,2))^2)/(2*r^2));
    end
    theta=(q2x'*w*q2x)\(q2x'*w*q2y);
    q_y(k)=theta(1)+theta(2)*q_x(k);
end
plot(q_x,q_y);

legend('trainingdata',strcat('r = ',num2str(r)));

 

posted @ 2018-06-12 10:02  ZinYY  阅读(1646)  评论(0编辑  收藏  举报