pytorch線性回歸的便捷實現

繼上一篇blog，使用更加簡潔的方法實現線性回歸
生成數據集：

num_inputs = 2 num_examples = 1000 true_w = [2, -3.4] true_b = 4.2 features = torch.tensor(np.random.normal(0, 1, (num_examples, num_inputs)), dtype=torch.float) labels = true_w[0] * features[:, 0] + true_w[1] * features[:, 1] + true_b labels += torch.tensor(np.random.normal(0, 0.01, size=labels.size()), dtype=torch.float)

讀取數據：
與上一篇blog實現的方式不同，這里使用DataLoader()創建數據集

import torch.utils.data as Databatch_size = 10 # 將訓練數據的特征和標簽組合 dataset = Data.TensorDataset(features, labels) # 隨機讀取小批量 data_iter = Data.DataLoader(dataset, batch_size, shuffle=True)

定義模型：

class LinearNet(nn.Module):def __init__(self, n_feature):super(LinearNet, self).__init__()self.linear = nn.Linear(n_feature, 1) # 使用nn.Linear自動創建線性規劃模型# forward 定義前向傳播def forward(self, x):y = self.linear(x)return ynet = LinearNet(num_inputs) print(net) # 使用print可以打印出網絡的結構

輸出網結構：

LinearNet((linear): Linear(in_features=2, out_features=1, bias=True) )

還可以用nn.Sequential來更加方便地搭建網絡，Sequential是一個有序的容器，網絡層將按照在傳入Sequential的順序依次被添加到計算圖中

# 寫法一 net = nn.Sequential(nn.Linear(num_inputs, 1)# 此處還可以傳入其他層)# 寫法二 net = nn.Sequential() net.add_module('linear', nn.Linear(num_inputs, 1)) # net.add_module(.....)# 寫法三 from collections import OrderedDict net = nn.Sequential(OrderedDict([('linear', nn.Linear(num_inputs, 1))# ......]))print(net) print(net[0])

輸出：

Sequential((linear): Linear(in_features=2, out_features=1, bias=True) ) Linear(in_features=2, out_features=1, bias=True)

通過net.parameters()來查看模型所有的可學習參數，此函數將返回一個生成器

for param in net.parameters():print(param)

輸出：

Parameter containing: tensor([[-0.0277, 0.2771]], requires_grad=True) Parameter containing: tensor([0.3395], requires_grad=True)

初始化參數

from torch.nn import initinit.normal_(net[0].weight, mean=0, std=0.01) # 初始化第一個變量 init.constant_(net[0].bias, val=0) # 也可以直接修改bias的data: net[0].bias.data.fill_(0)

定義模型參數：

loss = nn.MSELoss() # 損失函數 import torch.optim as optim optimizer = optim.SGD(net.parameters(), lr=0.03) # 定義優化器 print(optimizer) # 調整學習率 for param_group in optimizer.param_groups:param_group['lr'] *= 0.1 # 學習率為之前的0.1倍 SGD ( Parameter Group 0dampening: 0lr: 0.03momentum: 0nesterov: Falseweight_decay: 0 )

訓練模型：

num_epochs = 3 for epoch in range(1, num_epochs + 1):for X, y in data_iter:output = net(X)l = loss(output, y.view(-1, 1))optimizer.zero_grad() # 梯度清零，等價于net.zero_grad()l.backward()optimizer.step()print('epoch %d, loss: %f' % (epoch, l.item()))

查看模型訓練的參數：

dense = net[0] print(true_w, dense.weight) print(true_b, dense.bias)

總結

以上是生活随笔為你收集整理的（pytorch-深度学习系列）pytorch线性回归的便捷实现的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。