8_深度学习_线性回归的简洁实现

在之前的描述中,我们只依赖了:

  1. 通过张量来进行数据存储和线性代数
  2. 通过自动微分来计算梯度

实际上,由于数据迭代器、损失函数、优化器和神经网络很常用,现代深度学习库也为我们实现了这些组件。

我们将进一步介绍如何通过使用深度学习框架来简洁地实现线性回归模型。

生成数据集

import numpy as np
import torch
from torch.utils import data
from d2l import torch as d2l
true_w = torch.tensor([2, -3.4])
true_b = 4.2
features, labels = d2l.synthetic_data(true_w, true_b, 1000)

读取数据集

我们可以调用框架中现有的API来读取数据。我们将features和labels作为API的参数传递,并在实例化数据迭代器对象时指定batch_size。此外,布尔值is_train表示是否希望数据迭代器对象在每个迭代周期内打乱数据。

def load_array(data_arrays, batch_size, is_train=True):
	"""构造一个PyTorch数据迭代器"""
	dataset = data.TensorDataset(*data_arrays)
	return data.DataLoader(dataset, batch_size, shuffle=is_train)
batch_size = 10
data_iter = load_array((features, labels), batch_size)

为了验证是否正常工作,让我们读取并打印第一个小批量样本。这里我们使用iter构造Python迭代器,并使用next从迭代器中获取第一项。

next(iter(data_iter))
[tensor([[-0.6054, -0.8844],
		 [-0.7336, 1.4959],
		 [ 0.0251, -0.5782],
		 [-0.3874, -2.7577],
  	     [-0.2987, -1.6454],
		 [-1.6794, 0.9353],
		 [-1.1240, 0.0860],
		 [ 0.7230, -0.3869],
		 [ 0.2812, -1.3614],
		 [ 1.4122, 0.7235]]),
 tensor([[ 5.9993],
		 [-2.3407],
		 [ 6.2172],
		 [12.8106],
		 [ 9.2043],
		 [-2.3530],
		 [ 1.6639],
		 [ 6.9765],
		 [ 9.3874],
		 [ 4.5664]])]

定义模型

在之前实现线性回归时,我们明确定义了模型参数变量,并编写了计算的代码,这样通过基本的线性代数运算得到输出。但是,如果模型变得更加复杂,而且当我们几乎每天都需要实现模型时,我们会想简化这个过程。

对于标准操作,我们可以使用框架预定义好的层。这使我们只需关注使用哪些层来构造模型,而不必关注层的细节。我们首先定义一个模型变量net,它是一个Sequential类的实例。Sequential类为串联在一起的多个层定义了一个容器。当给定输入数据,Sequential实例将数据传入到第一层,然后将第一层的输出作为第二层的输入,以此类推。在下面的例子中,我们的模型只包含一个层,因此实际上不需要Sequential。但是由于以后几乎所有的模型都是多层的,在这里使用Sequential会让我们熟悉标准的流水线。

回顾之前的单层网络架构,这一单层被称为全连接层,因为它的每一个输入都通过矩阵-向量乘法连接到它的每个输出。

8_深度学习_线性回归的简洁实现_第1张图片

在PyTorch中,全连接层在Linear类中定义。值得注意的是,我们将两个参数传递到nn.Linear中。第一个指定输入特征形状,即2,第二个指定输出特征形状,输出特征形状为单一标量,因此为1。

# 'nn'是神经网络的缩写

from torch import nn

net = nn.Sequential(nn.Linear(2, 1))

初始化模型参数

在使用net之前,我们需要初始化模型参数。如在线性回归模型中的权重和偏置。深度学习框架通常有预定义的方法来初始化参数。在这里,我们指定每个权重参数应该从均值为0、标准差为0.01的正态分布中随机采样,偏置参数将初始化为零。

正如我们在构造nn.Linear时指定输入和输出尺寸一样。现在我们直接访问参数以设定初始值。我们通过net[0]选择网络中的第一个图层,然后使用weight.data和bias.data方法访问参数。然后使用替换方法normal_和fill_来重写参数值。

net[0].weight.data.normal_(0, 0.01)
net[0].bias.data.fill(0)

定义损失函数

计算均方误差使用的是MSELoss类,也称为平方 L 2 L_{2} L2范数。默认情况下,它返回所有样本损失的平均值。

loss = nn.MSELoss()

定义优化算法

小批量随机梯度下降算法是一种优化神经网络的标准工具,PyTorch在optim模块中实现了该算法的许多变种。当我们实例化SGD实例时,我们要指定优化的参数(可通过net.parameters()从我们的模型中获得)以及优化算法所需的超参数字典。小批量随机梯度下降只需要设置lr值,这里设置为0.03。

trainer = torch.optim.SGD(net.parameters(), lr = 0.03)

训练

回顾一下:在每个迭代周期里,我们将完整遍历一次数据集(train_data),不停地从中获取一个小批量的输入和相应的标签。对于每一个小批量,我们会进行以下步骤:

  • 通过调用net(X)生成预测并计算损失l(正向传播)
  • 通过进行反向传播来计算梯度
  • 通过调用优化器来更新模型参数

为了更好的衡量训练效果,我们计算每个迭代周期后的损失,并打印它来监控训练过程。

num_epochs = 3
for epoch in range(num_epochs):
	for X, y in data_iter:
		l = loss(net(X), y)
		trainer.zero_grad()
		l.backward()
		trainer.step()
	l = loss(net(features), labels)
	print(f'epoch {epoch + 1}, loss {l:f}')
epoch 1, loss 0.000254
epoch 2, loss 0.000098
epoch 3, loss 0.000098

下面我们比较生成数据集的真实参数和通过有限数据训练获得的模型参数。要访问参数,我们首先从net访问所需的层,然后读取该层的权重和偏置。正如在从零开始实现中一样,我们估计得到的参数与生成数据的真实参数非常接近。

w = net[0].weight.data
print('w的估计误差:', true_w - w.reshape(true_w.shape))
b = net[0].bias.data
print('b的估计误差', true_b - b)
w的估计误差: tensor([0.0007, 0.0007])
b的估计误差: tensor([-0.0007])

你可能感兴趣的:(pytorch,深度学习,python)