% 最小二乘法多項(xiàng)式擬合
% 原三次函數(shù)+隨機(jī)噪聲
f=@(x)x.^3+6.*x.^2-2.*x+4+(rand(size(x))-.5).*20;
% 構(gòu)造原始數(shù)據(jù)
x=-5:.1:5;
y=f(x);
% 自己寫一個(gè)最小二乘
n=3;% 最高次數(shù)為三次
X=(x').^(n:-1:0);
theta1=((X'*X)\X'*y')';
% MATLAB自帶多項(xiàng)式擬合
theta2=polyfit(x,y,n);
% 輸出對(duì)比
disp(theta1)
disp(theta2)
% 一個(gè)小技巧，下面的寫法能夠快速將
% 參數(shù)向量變成有關(guān)x的多項(xiàng)式匿名函數(shù)
func=matlabFunction(poly2sym(theta1));

% 繪圖部分
% 保持坐標(biāo)區(qū)域不刷新并添加網(wǎng)格
ax=gca;hold(ax,'on');grid(ax,'on');
% 繪制原數(shù)據(jù)點(diǎn)和擬合結(jié)果
plot(x,y,'o','MarkerFaceColor',[94,142,179]./255);
plot(x,func(x),'Color',[0,64,115]./255,'LineWidth',2);
% 修飾一下
ax.FontName='cambria';
ax.LineWidth=1.5;
ax.GridLineStyle='--';
ax.XColor=[1,1,1].*.3;
ax.YColor=[1,1,1].*.3;
ax.ZColor=[1,1,1].*.3;

% 最小二乘法多元線性回歸
% 原二元函數(shù)+隨機(jī)噪聲
f=@(x1,x2) 3.*x1+4.*x2+5+(rand(size(x1))-.5).*10;
% 構(gòu)造原始數(shù)據(jù)
[x1,x2]=meshgrid(-5:.5:5,-5:.5:5);
y=f(x1,x2);
% 自己寫一個(gè)最小二乘
X=[x1(:),x2(:),ones(size(x1(:)))];
theta1=((X'*X)\X'*y(:));
% MATLAB多元線性回歸
theta2=regress(y(:),X);
% 輸出對(duì)比
disp(theta1)
disp(theta2)
% 構(gòu)造擬合結(jié)果的二元匿名函數(shù)
func=matlabFunction([sym('x1'),sym('x2'),1]*theta1);

% 繪圖部分
% 保持坐標(biāo)區(qū)域不刷新并添加網(wǎng)格
ax=gca;hold(ax,'on');grid(ax,'on');
% 繪制原數(shù)據(jù)點(diǎn)和擬合結(jié)果
mesh(x1,x2,func(x1,x2),'FaceColor','flat','FaceAlpha',.8)
scatter3(x1(:),x2(:),y(:),20,'filled')
% 修飾一下
ax.FontName='cambria';
ax.LineWidth=1.5;
ax.GridLineStyle='--';
ax.XColor=[1,1,1].*.3;
ax.YColor=[1,1,1].*.3;
ax.ZColor=[1,1,1].*.3;
view(30,20)