#04 构建您的第一个神经网络：PyTorch入门指南

文章目录

前言
- 理论基础
- - 神经网络层的组成
  - 前向传播与反向传播
- 神经网络设计
- - 步骤1：准备数据集
  - 步骤2：构建模型
  - 步骤3：定义损失函数和优化器
  - 步骤4：训练模型
  - 步骤5：评估模型
  - 结论

前言

在过去的几天里，我们深入了解了深度学习的基础，从安装PyTorch开始，到理解Tensor运算，以及自动微分系统Autograd的工作原理。今天，我们将整合我们所学的知识和技能，并迈出一个重要的步伐：构建和训练我们的第一个神经网络。
在这里插入图片描述

在本篇博文中，我会指导您如何设计一个简单的全连接神经网络（也称为多层感知机MLP），并使用PyTorch作为您的工具。我们的目标是用这个网络解决一个分类问题，并了解神经网络训练的基本流程。让我们一步一步地进行。

理论基础

在开始编码之前，让我们先回顾一下全连接神经网络的基本组件和原理。一个典型的神经网络由多个层组成，每个层由多个神经元组成。在全连接网络中，一个层中的每个神经元都与前一层的所有神经元连接。这种密集的连接模式能让网络从输入数据中捕获复杂的模式。

神经网络层的组成

神经网络中的每一层主要包含以下部分：

输入节点：这些是将数据输入到网络的节点。
权重：每个连接都有一个权重，它决定了输入信号对神经元的激活程度的影响。
偏置：每个神经元都有一个偏置，它提供了除了输入以外的另一个调节激活的手段。
激活函数：它决定了神经元的输出，通常是一个非线性函数，如ReLU或Sigmoid。

前向传播与反向传播

神经网络的训练分为两个阶段：前向传播和反向传播。

前向传播：在这个阶段，输入数据通过网络的每一层，每个神经元根据其权重、偏置和激活函数计算输出。
反向传播：训练的这一部分涉及根据网络输出和实际结果之间的差异来调整权重和偏置。这是通过计算损失函数的梯度并将其反向传播到网络中完成的。

神经网络设计

现在，让我们设计一个简单的神经网络。假设我们要解决的是一个二分类问题，我们的网络将有两个输入节点（对应于两个特征），几个隐藏层，以及一个输出节点。

步骤1：准备数据集

在构建模型之前，我们首先需要准备数据集。通常，我们会对数据进行预处理，如标准化，然后分为训练集和测试集。

import torch
from torch.utils.data import DataLoader, TensorDataset
from sklearn.model_selection import train_test_split

# 假设data为特征，labels为标签
data_train, data_test, labels_train, labels_test = train_test_split(data, labels, test_size=0.2)
train_dataset = TensorDataset(torch.tensor(data_train, dtype=torch.float32), torch.tensor(labels_train, dtype=torch.float32))
test_dataset = TensorDataset(torch.tensor(data_test, dtype=torch.float32), torch.tensor(labels_test, dtype=torch.float32))

train_loader = DataLoader(dataset=train_dataset, batch_size=64, shuffle=True)
test_loader = DataLoader(dataset=test_dataset, batch_size=64, shuffle=False)

步骤2：构建模型

接下来，我们将使用PyTorch定义我们的神经网络。我们将使用nn.Module作为基类，并定义我们的层和前向传播方法。

import torch.nn as nn
import torch.nn.functional as F

class SimpleNN(nn.Module):
    def __init__(self):
        super(SimpleNN, self).__init__()
        self.fc1 = nn.Linear(2, 5)
        self.fc2 = nn.Linear(5, 1)

    def forward(self, x):
        x = F.relu(self.fc1(x))
        x = torch.sigmoid(self.fc2(x))
        return x

model = SimpleNN()

步骤3：定义损失函数和优化器

损失函数用于评估模型的预测与实际标签之间的差异。优化器用于根据损失函数的梯度更新模型的权重。

import torch.optim as optim

criterion = nn.BCELoss()
optimizer = optim.Adam(model.parameters(), lr=0.001)

步骤4：训练模型

训练模型涉及多个迭代（或“epoch”），在每个迭代中，我们将完成一个完整的前向和后向传播过程。

for epoch in range(100):  # 举例迭代100次
    for inputs, labels in train_loader:
        optimizer.zero_grad()
        outputs = model(inputs)
        loss = criterion(outputs.squeeze(), labels)
        loss.backward()
        optimizer.step()

步骤5：评估模型

最后，我们需要评估我们的模型性能，通常使用测试集来完成这个任务。

correct = 0
total = 0
with torch.no_grad():
    for inputs, labels in test_loader:
        outputs = model(inputs)
        predicted = outputs.round().squeeze()  # 将输出四舍五入为0或1
        total += labels.size(0)
        correct += (predicted == labels).sum().item()

print(f'Accuracy: {100 * correct / total}%')