【机器学习--MATLAB】基于MATAB利用线性模型进行拟合预测

题目

        1.如下图，基于线性模型拟合工资与额度的线性关系，并预测当工资为15000元时的额度是多少？

2.如下图，第1列是x1,第2列是x2,第3列是y，请用原始算法进行线性拟合并作图。

 

理论

 

 

 

代码

1.第1题

clear all
clc
%绘制散点图
data = xlsread('test1_data.xlsx');
x=data(:,1);
y=data(:,2);
scatter(x,y,'filled');
%绘制拟合图像   纯手动算法实现
m=length(x);
w=(sum(y*(x-mean(x))'))./(sum(x.^2)-(sum(x).^2)/m);
b=sum(y-w.*x)/m;
y=w*x+b;
hold on
plot(x,y);

clear all
clc
%绘制散点图
data = xlsread('test1_data.xlsx');
x=data(:,1);
y=data(:,2);
scatter(x,y,'filled');
%绘制拟合图像  1次多项式拟合(线性)
hold on
K=polyfit(x,y,1);
k1=K(1);
k2=K(2);
x_line=linspace(0,12000,10100);
y_line=k1*x_line+k2;
hold on
plot(x_line,y_line);
%预测
yuce_xdata = 15000;
yuce_ydata = k1*yuce_xdata+k2;

作图结果

 2.第2题

%%
clear all
clc
%绘制散点图
data = xlsread('test1_data_v2.xlsx');
x=data(:,1:2);
x1=data(:,1);
x2=data(:,2);
y=data(:,3);
y=y';
X=[x,ones(length(y),1)];
scatter3(x1,x2,y,'filled');
%绘制3维数据线性拟合图
w_=inv(X'*X)*X'*y';
x1=linspace(0,5,100);
x2=linspace(0,5,100);
yuce=w_(3)*x1+w_(2)*x2+w_(1);
for i=1:length(x1)
    for j=1:length(x2)
        yuce_juzhen(i,j)=yuce(i);
    end
end
hold on
mesh(x1,x2,yuce_juzhen);
%预测
yuce_x1data=7;yuce_x2data=6;
yuce_ydata=w_(3)*yuce_x1data+w_(2)*yuce_x2data+w_(1)

 

作图结果