邏輯回歸二分類
今天嘗試寫了一下邏輯回歸分類,把代碼分享給大家,至於原理的的話請戳這里 https://blog.csdn.net/laobai1015/article/details/78113214 (在這片博客的基礎上我加了一丟丟東西)。
用到的預測函數為

其中,h為預測函數(大於0.5為一類,小於等於0.5為另一類)。θ為各個特征的參數。θ=[θ1,θ2,θ3...]T
損失函數J(θ)為

利用梯度下降算法進行參數的更新公式如下:

其中,α是學習率參數,λ是正則項參數,需要自己輸入。
用上述三個公式即可寫出一般情況下的邏輯回歸分類,經試驗,其分類效果在我的測試數據下有90%以上准確度。
以下代碼為了演示方面,只提供了兩個特征值,然后加了一個常數項(經過實驗,常數項是不可缺少的)。大家在使用自己的數據集的話只需要進行一點小改動即可。
matlab代碼如下:
clear
clc
%% 數據准備
%X = xlsread('C:\Users\user01\Desktop\test.xlsx');
%二分類 隨機生成數據。 200個數據 每個數據2個特征
data=1*rand(300,2);
label=zeros(300,1);
%label(sqrt(data(:,1).^2+data(:,2).^2)<8)=1;
label((data(:,2)+data(:,1)>1))=1;
%在data上加常數特征項;
data=[data,ones(size(data,1),1)];
%打亂循序
randIndex = randperm(size(data,1));
data_new=data(randIndex,:);
label_new=label(randIndex,:);
%80%訓練 20%測試
k=0.8*size(data,1);
X1=data_new(1:k,:);
Y1=label_new(1:k,:);
X2=data_new(k+1:end,:);
Y2=label_new(k+1:end,:);
[m1,n1] = size(X1);
[m2,n2] = size(X2);
Features=size(data,2); %特征個數
%% 開始訓練
%設定學習率為0.01
delta=1;
lamda=0.2; %正則項系數
theta1=rand(1,Features);
%theta1=[.5,.5];
%%訓練模型
%梯度下降算法求解theta(每次都是對全部的數據進行訓練)
num = 300; %最大迭代次數
L=[];
while(num)
dt=zeros(1,Features);
loss=0;
for i=1:m1
xx=X1(i,1:Features);
yy=Y1(i,1);
h=1/(1+exp(-(theta1 * xx')));
dt=dt+(h-yy) * xx;
loss=loss+ yy*log(h)+(1-yy)*log(1-h);
end
loss=-loss/m1;
L=[L,loss];
theta2=theta1 - delta*dt/m1 - lamda*theta1/m1;
theta1=theta2;
num = num - 1;
if loss<0.01
break;
end
end
figure
subplot(1,2,1)
plot(L)
title('loss')
subplot(1,2,2)
x=0:0.1:10;
y=(-theta1(1)*x-theta1(3))/theta1(2);
plot(x,y,'linewidth',2)
hold on
plot(data(label==1,1),data(label==1,2),'ro')
hold on
plot(data(label==0,1),data(label==0,2),'go')
axis([0 1 0 1])
%測試數據
acc=0;
for i=1:m2
xx=X2(i,1:Features)';
yy=Y2(i);
finil=1/(1+exp(-theta2 * xx));
if finil>0.5 && yy==1
acc=acc+1;
end
if finil<=0.5 && yy==0
acc=acc+1;
end
end
acc/m2
%測試結果: 測試准確率為95%。
得到的loss曲線為:

分類圖像為:

