吳恩達深(shen)(shen)度學習(xi)(xi)課(ke)程一:神經網絡和深(shen)(shen)度學習(xi)(xi) 第二周:神經網絡基礎(chu) 課(ke)后習(xi)(xi)題和代碼實(shi)踐
此分類用于記錄吳恩達深度學習課程的學習筆記。
課程相關信(xin)息鏈(lian)接(jie)如下(xia):
- 原課程視頻鏈接:
- github課程資料,含課件與筆記:
- 課程配套練習(中英)與答案:
本篇為第一(yi)課(ke)第二周的(de)課(ke)程(cheng)習題部分的(de)講(jiang)解(jie)。
1.理論習題
這(zhe)是本周理論(lun)部分的習題(ti)和相應解析,博主已經(jing)做好了翻(fan)譯和解析工作,因此便不再(zai)重(zhong)復,但有一題(ti)涉(she)及到代(dai)碼中矩陣(zhen)的兩種乘法(fa),我們把(ba)這(zhe)道題(ti)單獨拿出看一下(xia):
看一下下面的代碼:
a = np.random.randn(3, 3)
b = np.random.randn(3, 1)
c = a * b
請問c的維度會是多少(shao)?
這便是這道題的內容,按照我們的學習的線代知識,我們應該會得到結果為 \((3,1)\) ,但實際上這道題的答案是 \((3,3)\)
這便涉(she)及到(dao)在python代碼中對矩陣的(de)兩(liang)類乘(cheng)法,以題(ti)里的(de)量為例:
c = a * b #Hadamard積(逐元素積)
c = a @ b #內積
這是兩種(zhong)不同(tong)的乘法,也會帶來不同(tong)結果,我(wo)們(men)用實際代(dai)碼證明(ming)一(yi)下:
import numpy as np
# 定義矩陣 a 和列向量 b
a = np.array([[1, 1, 1],
[1, 1, 1],
[1, 1, 1]])
b = np.array([[1],
[2],
[3]])
# 逐元素乘法 (Hadamard積)
c_hadamard = a * b
print("Hadamard積結果 (a * b):")
print(c_hadamard)
print("Hadamard積的維度:", c_hadamard.shape)
# 矩陣乘法 (內積)
c_dot = a @ b
print("\n矩陣乘法結果 (a @ b):")
print(c_dot)
print("矩陣乘法的維度:", c_dot.shape)
直接看結果:

現在來解釋一下,我們可以通過結果得知,我們在線代中學習的內積運算在代碼中被定義為 \(@\)
而直接使用 \(*\) 符號來運算,實際上是對 \(*\) 前的矩陣的每個元素分別做乘法。
結合廣播機制,我們可以知道 \(c = a * b\) 的實(shi)際運算過程是這樣的:
- 經過廣播機制, \(b\) 的內容以相同的形式向下復制兩行,讓維度和 \(a\) 一樣變為 \((3,3)\)
- \(a和b\) 的相同位置相乘得到結果放入 \(c\) 的相同位置,用公式來說就是:
現(xian)在我們(men)可以總結一(yi)下兩種乘(cheng)法的(de)區別如(ru)下:
- \(@\) 是線代中的內積乘法,當兩個矩陣的維度為\((m,n),(n,k)\) 即前一矩陣的列數等于后一矩陣的行數時才可進行運算,內積會改變矩陣的維度,結果矩陣的維度為 \((m,k)\)
- \(*\) 是逐個乘法,實際上是對 \(*\) 前的矩陣的每個元素分別做乘法,當兩個矩陣使用逐個乘法運算時,python會自動將后一矩陣廣播至前一矩陣大小,Hadamard積不會改變矩陣維度大小,結果矩陣和前一矩陣維度相同。
2.編程題:實現具有神經網絡思維的Logistic回歸
同樣,這是整理了課程習題的博主的編碼答案,其思路邏輯和可視化部分都非常完美,在不借助很多現在流行框架的情況下手動構建線性組合,激活函數等實現邏輯回歸。
因此我便不再重復,這里給出我的另一個版本以供參考,會更偏向于使用目前普遍使用的框架和內置函數來構建模型,偏向展示一個較為通用的模型訓練流程。
本次構建我們使用比較普遍的pytorch框架來實現這個模型,而配置pytorch環境在不熟悉的情況下可能會比較困難且繁瑣,如果希望動手實操但還沒有配置相關環境,這里我推薦按下面這位up主的視頻過程配置pytorch環境來進行練習:
后面的(de)課程筆記部分還(huan)會再(zai)介(jie)紹(shao)另(ling)一個主(zhu)流框架:tensorflow
2.1 數據準備
使用貓狗二分類數據集,其下載地址為:
數據集共2400幅圖像,其優點為貓狗各1200幅,做到了樣本均衡。
但缺點也存在,數據規模不大,且每幅圖像大小不一,需要我們進行一定的預處理。
我們(men)便以(yi)此數據集訓練一個可(ke)以(yi)分類貓狗(gou)的(de)Logistic回歸模型。
2.2 代碼邏輯
完整的代碼會附在文末
2.2.1 導入所需庫
我(wo)們來(lai)一個個看(kan)一看(kan)所需的庫,并介紹它們所起的作用。
(1)torch
import torch # PyTorch 主庫,提供張量計算和自動求導的核心功能
import torch.nn as nn # 神經網絡模塊,包含各種層和激活函數
import torch.optim as optim # 優化器模塊,用于參數更新
from torch.utils.data import DataLoader, random_split #數據加載與劃分工具
當我們看到這個導入格式時,可能會產生這樣一個疑問:
既然已經在第一行已經導入了torch庫,為什么我們還要再顯式地導入torch的子模塊?
對于這(zhe)個問題,我(wo)們先簡單解(jie)釋一下(xia)python的導庫(ku)語(yu)法(fa)。
- 直接導入庫
import torch #import 庫名:直接導入所需庫
#這樣導入后,我們就可以直接使用 torch.方法名(方法參數)來直接調用torch的方法。
x = torch.tensor([1, 2, 3]) #創建一個張量,張量同樣是庫定義的一種容納矩陣等內容的數據結構,類似之前的numpy庫。
#同時,如果torch存在子模塊,我們也可以用 torch.子模塊名.子模塊方法名(方法參數)的形式調用。
y = torch.nn.Sigmoid() #創建一個 Sigmoid 激活函數層,并把它賦值給變量 `y`。
- 逐層導入庫的子模塊
# 這種形式實際上是為了簡化代碼,先繼續看上部分:
import torch
y = torch.nn.Sigmoid() #這是直接導入的調用方式
import torch.nn as nn # import 庫.模塊名 as 別名
y = nn.Sigmoid() #這是導入子模塊后的調用方式
#我們可以直接用 as 后我們為這個模塊起的別名直接調用其方法。
#!注意,如果不使用 as 即:
import torch.nn #這種方法和 import torch 的調用方式上沒任何區別,相當于直接導入子模塊。
y = torch.nn.Sigmoid()
- 直接導入庫的子模塊或方法
#這樣其實也是進一步簡化代碼,也涉及打包時的優化
#按照之前的格式,我們先進行下面的導入
import torch.utils.data as data
loader = data.DataLoader(dataset, batch_size=32)
# DataLoader是 PyTorch 中用來加載數據的一個類,它能夠高效地處理數據集的批次(batch)加載,并且支持多種數據加載策略
# DataLoader(數據集,批次大小):會根據數據集 `dataset` 和指定的 `batch_size` 來加載數據,并返回一個可迭代的對象
#現在我們想進一步簡化,就是這種形式:
from torch.utils.data import DataLoader
# from 上層模塊 import 類或方法名
# 現在我們使用DataLoader的格式就是:
loader = DataLoader(dataset, batch_size=32)
以這(zhe)樣的(de)(de)幾種(zhong)導入(ru)格(ge)式,可(ke)以幫(bang)助我們實(shi)現較為精準(zhun)的(de)(de)導入(ru)使用(yong)的(de)(de)模塊或方(fang)法,實(shi)際上,這(zhe)也是目前普遍的(de)(de)使用(yong)方(fang)法,甚至(zhi)官方(fang)文檔(dang)中使用(yong)的(de)(de)也是這(zhe)種(zhong)格(ge)式而非(fei)直接導入(ru)整(zheng)個庫(ku)。
(2)torchvision
from torchvision import datasets, transforms
# torchvision是一個與PyTorch配合使用的開源計算機視覺工具庫,常用于計算機視覺領域的任務,如圖像分類、目標檢測、圖像分割等
# datasets模塊,包含了多種常用的計算機視覺數據集,自動下載并加載常見的數據集,并將其轉換為torch.utils.data.Dataset格式
# transforms用于對圖像進行預處理和數據增強。它提供了很多常見的圖像轉換操作,如縮放、裁剪、歸一化、旋轉等
(3)matplotlib
import matplotlib.pyplot as plt
# matplotlib是一個強大的繪圖庫,用于創建靜態、動態和交互式的可視化圖表。
(4) sklearn
from sklearn.metrics import accuracy_score
# sklearn 是進行機器學習任務時非常基礎且實用的庫。它支持從數據預處理到模型評估的全流程,涵蓋了各種機器學習任務。
# metrics這個模塊提供了許多用于評估機器學習模型性能的函數,特別是在分類任務中。
# accuracy_score用于計算分類模型的準確率(Accuracy),即正確預測的樣本占所有樣本的比例。
2.2.2 數據集預處理和劃分
如果把訓練完整的模型比作做菜,在導入所有所需庫后,我們已經具備了構建模型的所有“廚具”。
而數據集,就相當于我們的“原料”。
這里也補充一下劃(hua)(hua)分數據(ju)集的種類,一般(ban),我們會把數據(ju)集劃(hua)(hua)分為以下三部(bu)分:
- 訓練集:用來訓練模型的數據,幫助模型學習和調整參數。
- 驗證集:用來調整模型超參數和優化模型的性能。
- 測試集:用來最終評估模型表現的數據,檢查模型的泛化能力和真實世界的適應性。
注意,這里出現了一個之前沒有提到過的概念:超參數
超參(can)數(shu)是(shi)指在(zai)模型訓(xun)練(lian)開始之前由人工設(she)定的(de)、用(yong)于控制模型結構或學習過(guo)(guo)程(cheng)的(de)參(can)數(shu)。與(yu)通(tong)過(guo)(guo)反向傳播算法自動(dong)學習得到的(de)模型參(can)數(shu)(如(ru)權重、偏(pian)置)不同,超參(can)數(shu)不會在(zai)訓(xun)練(lian)過(guo)(guo)程(cheng)中被更新,而(er)是(shi)由研究者或工程(cheng)師在(zai)實驗(yan)中通(tong)過(guo)(guo)經驗(yan)、搜(sou)索(suo)或驗(yan)證(zheng)集性能調優得到。
常見的超參數包括學(xue)習率、批(pi)大小、訓(xun)練輪數、網(wang)絡(luo)層數、正則化系數、激(ji)活函數類型(xing)等,這些(xie)我們定義的量是通過在驗證(zheng)集上的結果來不斷(duan)調優的。
繼續用做菜的例子來說的話,訓練集讓模型“學會做菜”, 驗證集讓模型“做得更好吃”, 測試集讓我們知道“這道菜到底好不好吃”。
補充了一些基礎知識后,我們現在繼續實操部分,先看一下數據集:


可以看到,我只是用兩個文件夾分別存放兩類的圖像,并沒有在這里就劃分訓練集,驗證集和測試集。
而且還有一個問題,那就是圖片的大小不一,深度學習模型要求每個輸入樣本的尺寸、通道數完全一致,如果輸入圖片大小不一(yi)致(zhi),很(hen)多操(cao)作無(wu)法(fa)統一(yi)處(chu)理,導(dao)致(zhi)特征(zheng)提取混亂。
網上對構建模型有這樣一個戲稱:賽博接水管,不(bu)無道(dao)理(li),我們在輸入(ru)前,一定要對樣本(ben)進行預處理(li),處理(li)完的(de)樣本(ben)要和“輸入(ru)水管”的(de)管口嚴(yan)絲合(he)縫,不(bu)能(neng)(neng)大也(ye)不(bu)能(neng)(neng)小,而具體訓練(lian)過程中也(ye)有其他體現。
Pytorch自(zi)然提供了相應(ying)方法:
# transforms.Compose的作用是將多個圖像變換操作串聯起來,形成一個“流水線”,這樣每張圖片在被加載時會依次執行這些操作。
transform = transforms.Compose([
transforms.Resize((128, 128)), # 統一圖片尺寸為(128,128)
transforms.ToTensor() # 轉為Tensor張量 [通道數,高,寬]
transforms.Normalize((0.5,), (0.5,)) # 標準化,把像素映射到(-1,1)防止梯度消失
])
#我們可以把transforms.Compose的返回值transform看作一個函數,輸入圖片,輸出圖片經過這些處理后的結果。
# ToTensor()這一步不能省略,因為pytorch框架只接受它定義的張量結構作為輸入,我們要初步使用框架,就要遵守它定義的規則。
#沒有 Normalize,輸入在 [0,1],非常大維度的累加后,Linear 輸出可能很大導致輸出接近于1,從而讓梯度消失。
這樣,我們就完成了預處理部分。要說明的是,這里我們僅做了一個大小統一,轉化為張量和標準化的操作,在實際訓練中,可能還有更多預處理的內容,我們遇到再說。
一般來說,預處理代碼就是跟在導庫后的第一部分內容。
在(zai)其之(zhi)后的下一步,就是(shi)載入和劃(hua)分數據集,來(lai)繼續(xu)看:
# 加載整個數據集
dataset = datasets.ImageFolder(root='./cat_dog', transform=transform)
# ImageFolder是一個圖像數據集加載器,能夠自動讀取一個按照文件夾結構分類的圖像數據集。
# 兩個參數 root即為數據集存放文件夾路徑,transform即為剛剛的預處理函數,在這里作為參數自動應用于加載的每個樣本。
# ImageFolder會自動讀取文件夾,并以文件夾名作為分類標簽標注文件夾里的內容,這里就會自動將圖片分為貓狗兩類。
# 最后我們得到的返回值dataset是一個數據集對象,每一項都是(圖片的張量表示, 標簽)的形式
# 先設置各部分大小
total_size = len(dataset)
train_size = int(0.8 * total_size) # 80% 訓練集
val_size = int(0.1 * total_size) # 10% 驗證集
test_size = total_size - train_size - val_size # 10% 測試集
# 按設置好的比例隨機劃分
# 這里的隨機是指對某張圖分入哪部分的隨機,并非比例隨機
train_dataset, val_dataset, test_dataset = random_split(
dataset=dataset,
lengths=[train_size, val_size, test_size]
)
# random_split可以把一個數據集對象隨機拆分成若干個子數據集。
# 兩個參數 dataset即為待拆分數據集,lengths即為每個子數據集的大小
# 最后的返回train_dataset, val_dataset, test_dataset就是劃分好的訓練集,驗證集和測試集。
# 定義批量數據迭代器
# 這里干的實際上是我們之前說了很久的向量化。
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False)
test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False)
# DataLoader 三個參數,第一個即為要處理的數據集,batch_size是批次大小,就是我們向量化里的m
# shuffle=True 表示每個 epoch 開始前,會隨機打亂數據集的順序,防止模型按固定順序學習
# 最后我們得到三個迭代器,即可用于各部分的輸入,DataLoader 加載數據時,會把多個樣本堆疊成 batch,即[批次,通道數,高,寬]
要說明的一點是,我們之前在理論部分的講解中的一次處理所有樣本,就是把這里的batch_size設置為訓練集大小,而我們處理完整個訓練集一次是一個輪次(epoch)。
這樣,一個epoch就會進行一次傳播。
而實際上,batch_size通常小于訓練集大小,而一個批次(batch)便會傳播一次,因此,在這種情況下,一個epoch就會進行多次傳播。
2.2.3 模型構建
這便是最核(he)心的部分了,我(wo)們用一個類(lei)(lei)(lei)來(lai)實現模(mo)(mo)型的架構,如果類(lei)(lei)(lei)的規(gui)模(mo)(mo)較大(da),我(wo)們會單獨創建文(wen)件存放模(mo)(mo)型類(lei)(lei)(lei)。類(lei)(lei)(lei)定(ding)義一般會包含兩類(lei)(lei)(lei)方法:
- 初始化方法:這個模型里有什么(層級,激活函數等)
- 向前傳播方法:輸入進入模型后怎么走
來看代碼:
class LogisticRegressionModel(nn.Module):
# 類繼承自nn.Module,是 PyTorch 所有模型的基類
#初始化方法
def __init__(self):
super().__init__() #父類初始化,用于注冊子模塊等,涉及源碼,這里當成固定即可。
self.flatten = nn.Flatten() #把張量后三維展平為一維(通道C*高H*寬W)
self.linear = nn.Linear(128 * 128 * 3, 1) # 輸入是128x128x3,輸出1個加權和
# nn.Linear接受的是二維輸入[batch_size, features],這里是[32,128 * 128 * 3]
# 但Linear層不需要在參數里寫 batch 維度,它內部會自動處理批量輸入,只關心每個樣本的特征數和每個樣本輸出的維度,這也是廣播機制的應用。
self.sigmoid = nn.Sigmoid() #激活函數
#向前傳播方法
def forward(self, x):
# 現在,x的維度是[32,3,128,128]
x = self.flatten(x) #1.展平
# 現在,x的維度是[32,128 * 128 * 3]
x = self.linear(x) #2.過線性組合得到加權和
# 現在,x的維度是[32,1]
x = self.sigmoid(x) #3.過激活函數得到輸出
# 現在,x的維度是[32,1]
return x
定義(yi)完成模型類(lei)后,我們便(bian)可以(yi)將(jiang)其(qi)實例化并加(jia)以(yi)使(shi)用(yong)。
2.2.4 設備選擇
model = LogisticRegressionModel() #實例化模型
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") #自動選擇CPU或者GPU
model.to(device) #模型的所有參數(權重、偏置)都會存放在對應設備里。
我(wo)們都知道CPU就像計(ji)算機的(de)(de)“大腦”,但在深度學習(xi)的(de)(de)模(mo)型訓(xun)(xun)練(lian)領域中(zhong),反而GPU更(geng)常用(yong),尤其是(shi)在較(jiao)大的(de)(de)模(mo)型訓(xun)(xun)練(lian)中(zhong),GPU 是(shi)深度學習(xi)的(de)(de)“加速引(yin)擎”,它用(yong)大量并行核心,把神經網絡訓(xun)(xun)練(lian)和推(tui)理(li)中(zhong)重復、耗時的(de)(de)矩陣運(yun)算做得又快又高效(xiao),所謂的(de)(de)“租用(yong)服(fu)(fu)務器(qi)讓模(mo)型跑快點”,實際上就是(shi)利用(yong)服(fu)(fu)務器(qi)的(de)(de)較(jiao)先進的(de)(de)GPU。
2.2.5 損失函數和優化器
我們同樣使用(yong)內置函(han)數來定義(yi)這兩部分:
criterion = nn.BCELoss() # 二分類的交叉熵損失。
optimizer = optim.SGD(model.parameters(), lr=0.01) #梯度下降法,每個批次傳播一次。
# SGD 兩個參數 model.parameters()即為模型的所有參數:權重,偏置等。
# lr即為學習率。
2.2.6 訓練和驗證
這部分的邏輯較為復雜,主要思路就是遍歷輪次和批次來進行傳播,并記錄相應量用于后續畫圖。
# 定義訓練的總輪數
epochs = 10 # 表示訓練整個數據集的次數
train_losses = [] # 用于記錄每個epoch的訓練損失,便于可視化
val_accuracies = [] # 用于記錄每個epoch驗證集準確率,便于可視化
# 開始訓練循環,每個epoch表示遍歷完整個訓練集一次
for epoch in range(epochs):
model.train() # 設置模型為訓練模式
epoch_train_loss = 0 # 用于累計該epoch的總訓練損失
# 遍歷訓練集DataLoader,每次獲取一個batch
for images, labels in train_loader:
# images: Tensor, 形狀 [32, 3, 128, 128]
# labels: Tensor, 形狀 [32]
images, labels = images.to(device), labels.to(device).float().unsqueeze(1)
# .to(device): 將張量移動到GPU或CPU
# .float(): 將標簽轉為float類型,因為BCELoss要求輸入為浮點數
# .unsqueeze(1): 在第1維增加一維,使labels形狀變為 [32, 1],與輸出匹配
# 前向傳播:輸入圖片,計算模型預測輸出
outputs = model(images) # 調用模型的forward方法,返回 [32, 1]
#官方推薦這樣的形式,實際上相當于model.forward(images)
loss = criterion(outputs, labels) # 計算二分類交叉熵損失,輸出標量Tensor
# 反向傳播與參數更新
optimizer.zero_grad() # 清空上一次梯度,避免梯度累加
loss.backward() # 自動求梯度,計算每個參數的梯度
optimizer.step() # 根據梯度更新參數,完成一次優化步驟
# 累計損失,用于計算平均訓練損失
epoch_train_loss += loss.item()
# .item():將單元素Tensor轉為Python浮點數,便于記錄
# 計算該epoch的平均訓練損失
avg_train_loss = epoch_train_loss / len(train_loader)
train_losses.append(avg_train_loss) # 保存到列表,用于后續繪圖
# 驗證集評估準確率
model. # 設置模型為評估模式
val_true, val_pred = [], [] # 用于記錄驗證集真實標簽和預測標簽
with torch.no_grad(): # 禁用梯度計算,節省顯存和計算量
for images, labels in val_loader: # 遍歷驗證集
images = images.to(device) # 移動到GPU/CPU
outputs = model(images) # 前向傳播得到預測概率
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
# outputs.cpu().numpy(): 移動到CPU并轉為numpy數組
# numpy不用導入,這是PyTorch 內部實現的橋接接口
# 要用到scikit-learn、matplotlib 等庫計算或可視化,這些庫只接受 CPU 數據
# > 0.5: 將概率轉換為0/1預測
# .astype(int): 把布爾值 True/False 轉為整數 1/0
# .flatten(): 將二維數組展平成一維
val_pred.extend(preds) # 將預測結果加入列表
val_true.extend(labels.numpy()) # 將真實標簽加入列表
# 使用sklearn計算驗證集準確率
val_acc = accuracy_score(val_true, val_pred)
val_accuracies.append(val_acc) # 保存準確率,用于繪圖
# 打印該epoch的訓練損失和驗證集準確率
print(f"輪次: [{epoch+1}/{epochs}], 訓練損失: {avg_train_loss:.4f}, 驗證準確率: {val_acc:.4f}")
# {變量:.4f}表示保留4位小數
2.2.7 可視化
plt.rcParams['font.sans-serif'] = ['SimHei']
# 設置全局字體為黑體(SimHei),支持中文顯示
# plt.rcParams 是 Matplotlib 的全局參數字典,可修改默認樣式
plt.rcParams['axes.unicode_minus'] = False
# 設置 False 表示允許正常顯示負號
# 繪制曲線
plt.plot(train_losses, label='訓練損失')
# 繪制訓練集損失曲線
# plt.plot(y, label=...) 用于繪制折線圖,label 用于圖例說明
plt.plot(val_accuracies, label='驗證準確率')
# 作繪制驗證集準確率曲線
# 設置標題與坐標軸
plt.title("訓練損失與驗證準確率隨輪次變化圖") # 設置圖表標題
plt.xlabel("訓練輪次(Epoch)") # 設置橫軸標題
plt.ylabel("數值") # 設置縱軸標題
plt.legend() # 顯示圖例,用于區分不同折線
# plt.legend() 會顯示各 plt.plot() 的 label 內容
plt.grid(True) # 開啟網格顯示
plt.show() # 顯示繪制的圖形窗口
2.2.8 最終測試
在最后用測試集進行評估之前,其實應該有根據訓練集對超參數進行調優的過程,但由于目前的篇幅已經較長了,我們先看完流程,我會在最后再附上一個使用方格調優版本的代碼。
# 模型評估(測試集)
model.
y_true, y_pred = [], []
# 定義兩個空列表,用于存儲測試集的真實標簽與預測標簽
with torch.no_grad():
for images, labels in test_loader:
images = images.to(device)
outputs = model(images)
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
y_pred.extend(preds)
y_true.extend(labels.numpy())
acc = accuracy_score(y_true, y_pred)
print(f"測試準確率: {acc:.4f}")
可以發現,驗證集和測試集的代碼部分幾乎沒有差別,二者的主(zhu)要差(cha)別在(zai)于它們起到的作用上。
2.3 結果分析

這是這個模型的結果,可以發現準確率不高,只有一半左右。
而造成這個結果的因素是多(duo)樣的:
- 數據集規模不大,圖片大小不一。
- 預處理簡單,可能造成失真。
- 模型結構簡單,擬合能力不強
- 沒有對超參數進行較詳細地調優
此外,還有其他影響因素。
而下面便是可視化部分的代碼結果:

我們可能會發現這樣一個問題:按照梯度下降法的思路,損失值應該一直下降才對,為什么反而會有升高的反復現象?
其實,這是一個非常常見的現象。
雖然梯度下降的理論目標是不斷讓損失函數下降,但在實際訓練中,損失值并不會嚴格單調遞減,原(yuan)因主要(yao)有(you)以下(xia)幾點:
- 每次迭代并不是用全部數據計算梯度,而是用一個小批量,不同批次的數據分布略有差異,會導致梯度方向有波動,因此損失可能短暫上升。
- 如果學習率偏大,每次更新的步長過長,可能會“越過”最優點,使損失出現震蕩。
- 即使是簡單模型,在高維空間中損失函數也可能存在多個局部極小值和鞍點,訓練過程可能會在這些區域間來回波動。
換句話說,總體趨勢下降才是關鍵,出現輕微的上升是正常現象,不代表模型沒有學習。
我們本篇的主要目的還是展示模型構建的過程,在之后的課程學習里,會涉及更多更復雜的算法,函數與優化等,我們到時使用其再來試試在貓狗二分類數據集上的分類效果。
最(zui)后,完整代碼(ma)如下:
- 示例版本:
import torch
import torch.nn as nn
import torch.optim as optim
from torchvision import datasets, transforms
from torch.utils.data import DataLoader, random_split
import matplotlib.pyplot as plt
from sklearn.metrics import accuracy_score
transform = transforms.Compose([
transforms.Resize((128, 128)),
transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,))
])
dataset = datasets.ImageFolder(root='./cat_dog', transform=transform)
train_size = int(0.8 * len(dataset))
val_size = int(0.1 * len(dataset))
test_size = len(dataset) - train_size - val_size
train_dataset, val_dataset, test_dataset = random_split(dataset, [train_size, val_size, test_size])
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False)
test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False)
class LogisticRegressionModel(nn.Module):
def __init__(self):
super().__init__()
self.flatten = nn.Flatten()
self.linear = nn.Linear(128 * 128 * 3, 1) #
self.sigmoid = nn.Sigmoid()
def forward(self, x):
x = self.flatten(x)
x = self.linear(x)
x = self.sigmoid(x)
return x
model = LogisticRegressionModel()
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
criterion = nn.BCELoss()
optimizer = optim.SGD(model.parameters(), lr=0.01)
epochs = 10
train_losses = []
val_accuracies = []
for epoch in range(epochs):
model.train()
epoch_train_loss = 0
for images, labels in train_loader:
images, labels = images.to(device), labels.to(device).float().unsqueeze(1)
outputs = model(images)
loss = criterion(outputs, labels)
optimizer.zero_grad()
loss.backward()
optimizer.step()
epoch_train_loss += loss.item()
avg_train_loss = epoch_train_loss / len(train_loader)
train_losses.append(avg_train_loss)
model.
val_true, val_pred = [], []
with torch.no_grad():
for images, labels in val_loader:
images = images.to(device)
outputs = model(images)
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
val_pred.extend(preds)
val_true.extend(labels.numpy())
val_acc = accuracy_score(val_true, val_pred)
val_accuracies.append(val_acc)
print(f"輪次: [{epoch+1}/{epochs}], 訓練損失: {avg_train_loss:.4f}, 驗證準確率: {val_acc:.4f}")
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
plt.plot(train_losses, label='訓練損失')
plt.plot(val_accuracies, label='驗證準確率')
plt.title("訓練損失與驗證準確率隨輪次變化圖")
plt.xlabel("訓練輪次(Epoch)")
plt.ylabel("數值")
plt.legend()
plt.grid(True)
plt.show()
model.
y_true, y_pred = [], []
with torch.no_grad():
for images, labels in test_loader:
images = images.to(device)
outputs = model(images)
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
y_pred.extend(preds)
y_true.extend(labels.numpy())
acc = accuracy_score(y_true, y_pred)
print(f"測試準確率: {acc:.4f}")
- 加入方格搜索優化超參數的版本:
import torch
import torch.nn as nn
import torch.optim as optim
from torchvision import datasets, transforms
from torch.utils.data import DataLoader, random_split
from sklearn.metrics import accuracy_score
import itertools
transform = transforms.Compose([
transforms.Resize((128, 128)),
transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,))
])
dataset = datasets.ImageFolder(root='./cat_dog', transform=transform)
train_size = int(0.8 * len(dataset))
val_size = int(0.1 * len(dataset))
test_size = len(dataset) - train_size - val_size
train_dataset, val_dataset, test_dataset = random_split(dataset, [train_size, val_size, test_size])
class LogisticRegressionModel(nn.Module):
def __init__(self):
super().__init__()
self.flatten = nn.Flatten()
self.linear = nn.Linear(128 * 128 * 3, 1)
self.sigmoid = nn.Sigmoid()
def forward(self, x):
x = self.flatten(x)
x = self.linear(x)
x = self.sigmoid(x)
return x
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
param_grid = {
'lr': [0.01, 0.001],
'batch_size': [16, 32],
'num_epochs': [5, 10]
}
best_acc = 0
best_params = None
for lr, batch_size, num_epochs in itertools.product(param_grid['lr'], param_grid['batch_size'], param_grid['num_epochs']):
print(f"當前超參數: 學習率={lr}, 批次大小={batch_size}, 總輪次={num_epochs}")
train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False)
test_loader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False)
model = LogisticRegressionModel().to(device)
criterion = nn.BCELoss()
optimizer = optim.SGD(model.parameters(), lr=lr)
for epoch in range(num_epochs):
model.train()
epoch_loss = 0
for images, labels in train_loader:
images, labels = images.to(device), labels.to(device).float().unsqueeze(1)
outputs = model(images)
loss = criterion(outputs, labels)
optimizer.zero_grad()
loss.backward()
optimizer.step()
epoch_loss += loss.item()
avg_loss = epoch_loss / len(train_loader)
model.
y_val_true, y_val_pred = [], []
with torch.no_grad():
for images, labels in val_loader:
images = images.to(device)
outputs = model(images)
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
y_val_pred.extend(preds)
y_val_true.extend(labels.numpy())
val_acc = accuracy_score(y_val_true, y_val_pred)
print(f"輪次 {epoch+1}/{num_epochs}, 損失: {avg_loss:.4f}, 驗證準確率: {val_acc:.4f}")
if val_acc > best_acc:
best_acc = val_acc
best_params = {'lr': lr, 'batch_size': batch_size, 'num_epochs': num_epochs, 'model_state_dict': model.state_dict()}
print(f"\n最佳驗證準確率: {best_acc:.4f} 超參數設置: {best_params}")
best_model = LogisticRegressionModel().to(device)
best_model.load_state_dict(best_params['model_state_dict'])
best_model.
y_test_true, y_test_pred = [], []
with torch.no_grad():
for images, labels in test_loader:
images = images.to(device)
outputs = best_model(images)
preds = (outputs.cpu().numpy() > 0.5).astype(int).flatten()
y_test_pred.extend(preds)
y_test_true.extend(labels.numpy())
test_acc = accuracy_score(y_test_true, y_test_pred)
print(f"使用最優超參數的測試準確率: {test_acc:.4f}")

