1. 導(dǎo)入庫

機(jī)器學(xué)習(xí)的任務(wù)分為兩大類：分類和回歸

分類是對一堆目標(biāo)進(jìn)行識別歸類，例如貓狗分類、手寫數(shù)字分類等等

回歸是對某樣事物接下來行為的預(yù)測，例如預(yù)測天氣等等

這次我們要完成的任務(wù)是邏輯回歸，雖然名字叫做回歸，其實是個二元分類的任務(wù)

首先看看我們需要的庫文件

torch.nn 是專門為神經(jīng)網(wǎng)絡(luò)設(shè)計的接口

matplotlib 用來繪制圖像，幫助可視化任務(wù)

torch 定義張量，數(shù)據(jù)的傳輸利用張量來實現(xiàn)

optim 優(yōu)化器的包，例如SGD等

numpy 數(shù)據(jù)處理的包

2. 定義數(shù)據(jù)集

簡單說明一下任務(wù)，想在一個正方形的區(qū)域內(nèi)生成若干點(diǎn)，然后手工設(shè)計label，最后通過神經(jīng)網(wǎng)絡(luò)的訓(xùn)練，畫出決策邊界

假設(shè)：正方形的邊長是2，左下角的坐標(biāo)為（0，0），右上角的坐標(biāo)為（2，2）

然后我們手工定義分界線 y = x ，在分界線的上方定義為藍(lán)色，下方定義為紅色

2.1 生成數(shù)據(jù)

首先生成數(shù)據(jù)的代碼為

首先通過rand（0-1的均勻分布）生成200個點(diǎn)，并將他們擴(kuò)大2倍，x1代表橫坐標(biāo)，x2代表縱坐標(biāo)

然后定義一下分類，這里簡單介紹一下zip函數(shù)。

zip會將這里的a，b對應(yīng)打包成一對，這樣i對應(yīng)的就是（1，‘a’）,i[0] 對應(yīng)的就是1 2 3

再回到我們的代碼，因為我們要實現(xiàn)的是二元分類，所以我們定義兩個不同的類型，用pos，neg存起來。然后我們知道i[1] 代表的是 x2 ，i[0] 代表的是x1 ， 所以 x2 - x1 <0 也就是也就是在直線y=x的下面為pos類型。否則，為neg類型

最后，我們需要將pos，neg類型的繪制出來。因為pos里面其實是類似于(1,1)這樣的坐標(biāo)，因為pos.append(i) 里面的 i 其實是(x1,x2) 的坐標(biāo)形式, 所以我們將pos 里面的第一個元素x1定義為賦值給橫坐標(biāo)，第二個元素x2賦值給縱坐標(biāo)

然后通過scatter 繪制離散的點(diǎn)就可以，將pos 繪制成 red 顏色，neg 繪制成 blue 顏色，如圖

2.2 設(shè)置label

我們進(jìn)行的其實是有監(jiān)督學(xué)習(xí)，所以需要label

這里需要注意的是，不同于回歸任務(wù)，x1不是輸入，x2也不是輸出。應(yīng)該x1，x2都是輸入的元素，也就是特征feature。所以我們應(yīng)該將紅色的點(diǎn)集設(shè)置一個標(biāo)簽，例如 1 ，藍(lán)色的點(diǎn)集設(shè)置一個標(biāo)簽，例如 0.

實現(xiàn)代碼如下

很容易理解，訓(xùn)練集x_data 應(yīng)該是所有樣本，也就是pos和neg的所以元素。而之前介紹了x1，x2都是輸入的特征，那么x_data的shape 應(yīng)該是 [200,2] 的。而y_data 只有1(pos 紅色)類別，或者 0(neg 藍(lán)色)類型，所以y_data 的shape 應(yīng)該是 [200,1] 的。y_data view的原因是變成矩陣的形式而不是向量的形式

這里的意思是，假如坐標(biāo)是(1.5,0.5)那么應(yīng)該落在紅色區(qū)域，那么這個點(diǎn)的標(biāo)簽就是1

3. 搭建網(wǎng)絡(luò)+優(yōu)化器

網(wǎng)絡(luò)的類型很簡單，不再贅述。至于為什么要繼承nn.Module或者super那步是干啥的不用管，基本上都是這樣寫的，記住就行。

需要注意的是我們輸入的特征是(n * 2) ，所以Linear 應(yīng)該是(2,1)

二元分類最后的輸出一般選用sigmoid函數(shù)

這里的損失函數(shù)我們選擇BCE，二元交叉熵?fù)p失函數(shù)。

算法為隨機(jī)梯度下降

4. 訓(xùn)練

訓(xùn)練的過程也比較簡單，就是將模型的預(yù)測輸出值和真實的label作比較。然后將梯度歸零，在反向傳播并且更新梯度。

5. 繪制決策邊界

這里模型訓(xùn)練完成后，將w0，w1 ，b取出來，然后繪制出直線

這里要繪制的是w0 * x1+ w1 * x2 + b = 0 ，因為最開始介紹了x1代表橫坐標(biāo)x，x2代表縱坐標(biāo)y。通過變形可知y = (-w0 * x1 - b ) / w1，結(jié)果如圖

程序輸出的損失為

最后，w0 = 4.1911 ， w1 = -4.0290 ，b = 0.0209 ，近似等于y = x，和我們剛開始定義的分界線類似

6. 代碼

import torch.nn as nn
import matplotlib.pyplot as plt
import torch
from torch import optim
import numpy as np
torch.manual_seed(1)    # 保證程序隨機(jī)生成數(shù)一樣
x1 = torch.rand(200) * 2
x2 = torch.rand(200) * 2
data = zip(x1,x2)
pos = []                # 定義類型 1
neg = []                # 定義類型 2
def classification(data):
    for i in data:
        if(i[1] - i[0] < 0):
            pos.append(i)
        else:
            neg.append(i)
classification(data)
pos_x = [i[0] for i in pos]
pos_y = [i[1] for i in pos]
neg_x = [i[0] for i in neg]
neg_y = [i[1] for i in neg]
plt.scatter(pos_x,pos_y,c='r')
plt.scatter(neg_x,neg_y,c='b')
plt.show()
x_data = [[i[0],i[1]] for i in pos]
x_data.extend([[i[0],i[1]] for i in neg])
x_data = torch.Tensor(x_data)     # 輸入數(shù)據(jù) feature
y_data = [1 for i in range(len(pos))]
y_data.extend([0 for i in range(len(neg))])
y_data = torch.Tensor(y_data).view(-1,1)    # 對應(yīng)的標(biāo)簽
class LogisticRegressionModel(nn.Module):   # 定義網(wǎng)絡(luò)
    def __init__(self):
        super(LogisticRegressionModel,self).__init__()
        self.linear = nn.Linear(2,1)
        self.sigmoid = nn.Sigmoid()
    def forward(self,x):
        x = self.linear(x)
        x = self.sigmoid(x)
        return x
model = LogisticRegressionModel()
criterion = nn.BCELoss()    
optimizer = optim.SGD(model.parameters(),lr =0.01)
for epoch in range(10000):
    y_pred = model(x_data)
    loss = criterion(y_pred,y_data)     # 計算損失值
    if epoch % 1000 == 0:
        print(epoch,loss.item())     # 打印損失值
    optimizer.zero_grad()         # 梯度清零
    loss.backward()                 # 反向傳播
    optimizer.step()                # 梯度更新
w = model.linear.weight[0]      # 取出訓(xùn)練完成的結(jié)果
w0 = w[0]
w1 = w[1]
b = model.linear.bias.item()
with torch.no_grad():      # 繪制決策邊界，這里不需要計算梯度
    x= torch.arange(0,3).view(-1,1)
    y = (- w0 * x - b) / w1
    plt.plot(x.numpy(),y.numpy())
plt.scatter(pos_x,pos_y,c='r')          
plt.scatter(neg_x,neg_y,c='b')
plt.xlim(0,2)
plt.ylim(0,2)
plt.show()

程序結(jié)果

到此這篇關(guān)于pytorch邏輯回歸實現(xiàn)步驟詳解的文章就介紹到這了,更多相關(guān)pytorch邏輯回歸內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論