PyTorch搭建一維線性回歸模型（二）

更新時間：2021年04月09日 14:27:04 作者：Liam Coder

這篇文章主要為大家詳細介紹了PyTorch搭建一維線性回歸模型，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

PyTorch基礎入門二：PyTorch搭建一維線性回歸模型

1）一維線性回歸模型的理論基礎

給定數據集，線性回歸希望能夠優(yōu)化出一個好的函數，使得能夠和盡可能接近。

如何才能學習到參數和呢？很簡單，只需要確定如何衡量與之間的差別，我們一般通過損失函數（Loss Funciton)來衡量：。取平方是因為距離有正有負，我們于是將它們變?yōu)槿钦摹＿@就是著名的均方誤差。我們要做的事情就是希望能夠找到和，使得：

均方差誤差非常直觀，也有著很好的幾何意義，對應了常用的歐式距離?，F在要求解這個連續(xù)函數的最小值，我們很自然想到的方法就是求它的偏導數，讓它的偏導數等于0來估計它的參數，即：

求解以上兩式，我們就可以得到最優(yōu)解。

2）代碼實現

首先，我們需要“制造”出一些數據集：

import torch
import matplotlib.pyplot as plt
 
 
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y = 3*x + 10 + torch.rand(x.size())
# 上面這行代碼是制造出接近y=3x+10的數據集，后面加上torch.rand()函數制造噪音
 
# 畫圖
plt.scatter(x.data.numpy(), y.data.numpy())
plt.show()

我們想要擬合的一維回歸模型是。上面制造的數據集也是比較接近這個模型的，但是為了達到學習效果，人為地加上了torch.rand()值增加一些干擾。

上面人為制造出來的數據集的分布如下：

有了數據，我們就要開始定義我們的模型，這里定義的是一個輸入層和輸出層都只有一維的模型，并且使用了“先判斷后使用”的基本結構來合理使用GPU加速。

class LinearRegression(nn.Module):
  def __init__(self):
    super(LinearRegression, self).__init__()
    self.linear = nn.Linear(1, 1) # 輸入和輸出的維度都是1
  def forward(self, x):
    out = self.linear(x)
    return out
 
if torch.cuda.is_available():
  model = LinearRegression().cuda()
else:
  model = LinearRegression()

然后我們定義出損失函數和優(yōu)化函數，這里使用均方誤差作為損失函數，使用梯度下降進行優(yōu)化：

criterion = nn.MSELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=1e-2)

接下來，開始進行模型的訓練。

num_epochs = 1000
for epoch in range(num_epochs):
  if torch.cuda.is_available():
    inputs = Variable(x).cuda()
    target = Variable(y).cuda()
  else:
    inputs = Variable(x)
    target = Variable(y)
 
  # 向前傳播
  out = model(inputs)
  loss = criterion(out, target)
 
  # 向后傳播
  optimizer.zero_grad() # 注意每次迭代都需要清零
  loss.backward()
  optimizer.step()
 
  if (epoch+1) %20 == 0:
    print('Epoch[{}/{}], loss:{:.6f}'.format(epoch+1, num_epochs, loss.data[0]))

首先定義了迭代的次數，這里為1000次，先向前傳播計算出損失函數，然后向后傳播計算梯度，這里需要注意的是，每次計算梯度前都要記得將梯度歸零，不然梯度會累加到一起造成結果不收斂。為了便于看到結果，每隔一段時間輸出當前的迭代輪數和損失函數。

接下來，我們通過model.eval()函數將模型變?yōu)闇y試模式，然后將數據放入模型中進行預測。最后，通過畫圖工具matplotlib看一下我們擬合的結果，代碼如下：

model.eval()
if torch.cuda.is_available():
  predict = model(Variable(x).cuda())
  predict = predict.data.cpu().numpy()
else:
  predict = model(Variable(x))
  predict = predict.data.numpy()
plt.plot(x.numpy(), y.numpy(), 'ro', label='Original Data')
plt.plot(x.numpy(), predict, label='Fitting Line')
plt.show()

其擬合結果如下圖：

附上完整代碼：

# !/usr/bin/python
# coding: utf8
# @Time  : 2018-07-28 18:40
# @Author : Liam
# @Email  : luyu.real@qq.com
# @Software: PyCharm
#            .::::.
#           .::::::::.
#           :::::::::::
#         ..:::::::::::'
#        '::::::::::::'
#         .::::::::::
#      '::::::::::::::..
#         ..::::::::::::.
#        ``::::::::::::::::
#        ::::``:::::::::'    .:::.
#        ::::'  ':::::'    .::::::::.
#       .::::'   ::::   .:::::::'::::.
#      .:::'    ::::: .:::::::::' ':::::.
#      .::'    :::::.:::::::::'   ':::::.
#     .::'     ::::::::::::::'     ``::::.
#   ...:::      ::::::::::::'       ``::.
#   ```` ':.     ':::::::::'         ::::..
#            '.:::::'          ':'````..
#           美女保佑 永無BUG
 
import torch
from torch.autograd import Variable
import numpy as np
import random
import matplotlib.pyplot as plt
from torch import nn
 
 
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y = 3*x + 10 + torch.rand(x.size())
# 上面這行代碼是制造出接近y=3x+10的數據集，后面加上torch.rand()函數制造噪音
 
# 畫圖
# plt.scatter(x.data.numpy(), y.data.numpy())
# plt.show()
class LinearRegression(nn.Module):
  def __init__(self):
    super(LinearRegression, self).__init__()
    self.linear = nn.Linear(1, 1) # 輸入和輸出的維度都是1
  def forward(self, x):
    out = self.linear(x)
    return out
 
if torch.cuda.is_available():
  model = LinearRegression().cuda()
else:
  model = LinearRegression()
 
criterion = nn.MSELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=1e-2)
 
num_epochs = 1000
for epoch in range(num_epochs):
  if torch.cuda.is_available():
    inputs = Variable(x).cuda()
    target = Variable(y).cuda()
  else:
    inputs = Variable(x)
    target = Variable(y)
 
  # 向前傳播
  out = model(inputs)
  loss = criterion(out, target)
 
  # 向后傳播
  optimizer.zero_grad() # 注意每次迭代都需要清零
  loss.backward()
  optimizer.step()
 
  if (epoch+1) %20 == 0:
    print('Epoch[{}/{}], loss:{:.6f}'.format(epoch+1, num_epochs, loss.data[0]))
model.eval()
if torch.cuda.is_available():
  predict = model(Variable(x).cuda())
  predict = predict.data.cpu().numpy()
else:
  predict = model(Variable(x))
  predict = predict.data.numpy()
plt.plot(x.numpy(), y.numpy(), 'ro', label='Original Data')
plt.plot(x.numpy(), predict, label='Fitting Line')
plt.show()

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: