代码参考了知乎以及机器学习实战的内容,数据集为机器学习实战的testSet.txt
machine learning inaction - Ch05 - testSet.txt
可自行在GitHub搜索获得,此处不提供了。
matlab代码
clear;
clc;
% 逻辑回归
path='E:\Codes\Python\ML\testSet.txt';
[data,label]=loadDataSet(path);
[m,n]=size(data);
alpha=0.001;
theta=ones(n,1);% 3*1
# 循环次数
Loop=10000;
for i=1:Loop
P = sigmoid(data*theta);
error=loss(P,label);
theta=update_theta(alpha,data,error,theta);
end
disp(theta);
% 绘图
%1. 描点,不同分类不同样式
path='E:\Codes\Python\ML\testSet.txt';
dataSet=importdata(path);
[m,n]=size(dataSet); % m*n
x1=[];y1=[];
x2=[];y2=[];
for i=1:m
if dataSet(i,3)==0
x1(end+1)=dataSet(i,1);
y1(end+1)=