搭建一个简单的深度神经网络

YSRM

已于 2024-06-12 13:04:28 修改

阅读量758

点赞数 5

分类专栏：深度学习文章标签： dnn pytorch 人工智能

于 2024-06-12 13:04:12 首次发布

本文链接：https://blog.csdn.net/weixin_58628068/article/details/139622406

版权

深度学习专栏收录该内容

7 篇文章 0 订阅

订阅专栏

本博客实验环境为jupyter

一、引入所需要的库

torch库是核心，其中torch.nn 提供了搭建网络所需的所有组件，nn即神经网络。matplotlib类似与matlab，其中pyplot用于进行数据可视化，如绘制图表、曲线等。%matplotlib inline: 这是IPython（Jupyter Notebook）的魔法命令，用于在Notebook中直接显示Matplotlib绘制的图表，而不是弹出一个新窗口显示。

import torch 
import torch.nn as nn 
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt 
%matplotlib inline

# 展示高清图 
from matplotlib_inline import backend_inline #导入Matplotlib库中的backend_inline模块，用于控制图表的显示方式。
backend_inline.set_matplotlib_formats('svg') #设置Matplotlib图表的显示格式为SVG格式，SVG格式的图表在显示时具有高清晰度，适合用于展示精细的图形。

二、制作数据集

主要任务是读取数据集，划分为训练集和测试集，一定要随机划分。

读取的数据集中共760组数据，共8个输入特征，1个输出特征。

其中第一列是索引，从0开始，70%为训练集，30%为测试集。

#读取数据
df = pd.read_csv('Data.csv', index_col=0)#之前的pandas库中有介绍到，即df为读取后的对象，以第一列为索引    
arr = df.values #转化为numpy数组              
arr = arr.astype(np.float32)#转化为深度学习常用的单精度浮点类型    
ts = torch.tensor(arr)#转化为张量tensor         
ts = ts.to('cuda')#送到cuda设备上即gpu上计算             

# 划分训练集与测试集 
train_size = int(len(ts) * 0.7) #训练集的大小为百分之七十          
test_size = len(ts) - train_size #测试集的大小为百分之三十         
ts = ts[ torch.randperm( ts.size(0) ) , : ] #随机打乱数据集中样本的顺序    
train_Data = ts[ : train_size , : ] #将前百分之七十行给训练集       
test_Data = ts[ train_size : , : ]  #将百分之七十后的行给测试集

三、搭建神经网络

主要是构建DNN类，需要对python的类定义有较为深入的理解能力。

class DNN(nn.Module): #定义了一个名为DNN的PyTorch模型类，该类继承自nn.Module类，表示这是一个神经网络模型。
 
    def __init__(self): #定义了模型的初始化方法
        ''' 搭建神经网络各层 ''' 
        super(DNN,self).__init__() #调用父类的初始化方法，确保模型的其他部分也能够被正确初始化。
        self.net = nn.Sequential(            # 按顺序搭建各层 
            nn.Linear(8, 32), nn.Sigmoid(),   # 第1层：全连接层 ,是一个包含32个神经元的全连接层，输入特征数为8（表示输入数据维度为8），并使用Sigmoid激活函数。
            nn.Linear(32, 8), nn.Sigmoid(),   # 第2层：全连接层 ,是一个包含8个神经元的全连接层，输入特征数为32，同样使用Sigmoid激活函数。
            nn.Linear(8, 4), nn.Sigmoid(),    # 第3层：全连接层 ,是一个包含4个神经元的全连接层，输入特征数为8，同样使用Sigmoid激活函数。
            nn.Linear(4, 1), nn.Sigmoid()    # 第4层：全连接层 ,是一个包含1个神经元的全连接层，输入特征数为4，同样使用Sigmoid激活函数。这是模型的输出层。
        ) 
 
    def forward(self, x): 
        ''' 前向传播 ''' 
        y = self.net(x)    # 将输入数据x通过模型定义的神经网络结构self.net进行前向传播计算，得到输出y。
        return y        # y即输出数据
 model = DNN().to('cuda:0')    # 创建子类的实例，并搬到GPU上

这个代码可以当做模板，其中需要修改的部分为网络层的搭建，输入特征，中间层，输出特征一般都要为2的n次幂。

这就是该实例的各层。

四、训练网络

通过前向传播，反向传播等操作，本质上是不断调整权重和偏置。

loss_fn = nn.BCELoss(reduction='mean')#选择二元交叉熵损失函数作为模型的损失函数，其中reduction='mean'表示采用平均损失值作为最终的损失值。
learning_rate = 0.005    # 设置学习率 
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate) 
# 训练网络 
epochs = 5000     #设置训练的总轮数为5000。
losses = []        # 记录损失函数变化的列表 
 
# 给训练集划分输入与输出 
X = train_Data[ : , : -1 ]                # 前8列为输入特征 
Y = train_Data[ : , -1 ].reshape((-1,1))    # 后1列为输出特征
for epoch in range(epochs):     #对于每个epoch进行循环。
    Pred = model(X)              #通过模型进行一次前向传播，得到模型的预测结果Pred。
    loss = loss_fn(Pred, Y)        #计算模型预测结果与实际标签之间的损失值。
    losses.append(loss.item())     #将当前轮次的损失值记录到losses列表中。
    optimizer.zero_grad()        #清空上一轮的梯度信息。
    loss.backward()             #进行反向传播，计算梯度。
    optimizer.step()             #根据优化算法更新模型的参数，完成一轮训练。

#绘制损失函数随训练轮次变化的图像，用于可视化训练过程中损失值的变化。
Fig = plt.figure() 
plt.plot(range(epochs), losses) 
plt.ylabel('loss') 
plt.xlabel('epoch') 
plt.show()

生成结果为

可以发现随着训练的进行，loss开始减少。

五、测试网络

通过用训练好的模型对测试集进行测试，由于只有一个输出特征为0或者1，将大于0.5的置为1，小于0.5的置为0，可以类比成可能性从0到1。

# 测试网络 
# 给测试集划分输入与输出 
X = test_Data[ : , : -1 ]                
Y = test_Data[ : , -1 ].reshape((-1,1))    
with torch.no_grad():    #进入上下文管理器，表示接下来的计算不会被记录在计算图中，因此不会影响梯度的计算。
    Pred = model(X)     
    Pred[Pred>=0.5] = 1 
    Pred[Pred<0.5] = 0 
    correct = torch.sum( (Pred == Y).all(1) )    #统计预测正确的样本数，使用.all(1)表示在第1维度（即行）上进行比较，得到一个布尔张量，再进行求和操作。
    total = Y.size(0)   #获取试集样本总数。
    print(f'测试集精准度: {100*correct/total} %')

一般精准度得百分之八九十才合格哦，所以精度不高很有可能是训练集或者环境的问题，所以训练前一定要做好准备工作，因为训练一个模型要花费很久时间。