[Machine Learning]pytorch手搓一个神经网络模型

因为之前虽然写过一点点关于pytorch的东西，但是用的还是他太少了。

这次从头开始，尝试着搓出一个神经网络模型

（因为没有什么训练数据，所以最后的训练部分使用可能不太好跑起来的代码作为演示，如果有需要自己连上数据集合进行修改捏）

1.先阐述一下什么是神经网络块（block）

一般来说，我们之前遇到的一些神经网络，网络中是这样子的结构

net----> layer ----> neuron

而块的存在，就是给这样一个神经网络的整体做了一个封装操作，让神经网络能复合实现一些功能。

结构就变成了这个样子（图片来自D2l）

这样子，神经网络结构就变成了四层

block ----》 net ---》layer ---》neuron

这样子自然是可以使用诸如一些奇怪的方法，通过三层索引去进行调用什么的，不过这个我们到后面再说。先看一下如何构建一个块。

我们这里构建了一个类，这个类的计算方法实际就是实现了几个层的输入和输出，相当与封装了一个神经网络。

python 复制代码

class MLP(nn.Module):
    # 用模型参数声明层。这里，我们声明两个全连接的层
    def __init__(self):
        # 调用MLP的父类Module的构造函数来执行必要的初始化。
        # 这样，在类实例化时也可以指定其他函数参数，例如模型参数params（稍后将介绍）
        super().__init__()
        self.hidden = nn.Linear(20, 256)  # 隐藏层
        self.out = nn.Linear(256, 10)  # 输出层
    # 定义模型的前向传播，即如何根据输入X返回所需的模型输出
    def forward(self, X):
        # 注意，这里我们使用ReLU的函数版本，其在nn.functional模块中定义。
        return self.out(F.relu(self.hidden(X)))
        # 这个东西就相当与先隐藏层，然后relu，然后最后进行一次输出

#创建这个神经网路块，然后开始输出
net = MLP()
net(X)

这段代码没有使用squential容器进行封装，但是可以很清楚地看到我们定义了两层（隐藏层256个神经元，输出层10个神经元，不知道为什么没用softmax函数），并且在返回函数计算的时候，中间还经过了一步'relu'激活函数的操作

（注意和tf不同，pytorch框架下面是不能把激活函数存入层中的，需要单独作为一个'层'来进行一个输入和输出的控制）

注意下（在后面自定义层的时候也是这样子）由于继承了nn.Module这个类 , 所以我们必须要实现两个函数，首先是_init_,这个在python中是最终要的构造函数。其次就是forward，我对py不是很了解，不过这应该是通过面向对象实现的集成。forward这个方法就是向前传播，也就是接受参数，内部计算，然后返回值传递下去。

我们直接给net对象传递我们随机生成的两条数据的时候，底层时就调用了这个函数。

其他的一些比如sequential的实现方法，在这里我们就不加以赘述了。

为了更好的解释forward这个函数的作用，在这里我们自己创建一个单层，通过类创建，仍然是获取一个集成nn.Module的类，然后内部设置好初始化（为了创建对象），设置好向前传播（为了用来调用）

python 复制代码

# 自定义一个不需要参数的层
class CenteredLayer(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, X):    #该层向前传播的方法
        return X - X.mean()
# 这一层最终也是返回一个张量
# sequential是一个简单的线性封装容器，所以只要是符合输入张量，输出张量
# 并且在内部会调用他们的forward方法

layer = CenteredLayer()
print('自定义层，每个元素都 - 平均值2',layer(torch.FloatTensor([1, 2, 3, 4, 5])))

这个单层的效果就是对每个元素，都减去平均值。

并且如果想的话，我们也可以创建一些拥有自己属性的层

python 复制代码

#现在创建一个带有权重和偏好
class MyLinear(nn.Module):
    def __init__(self, in_units, units):
        super().__init__()
        self.weight = nn.Parameter(torch.randn(in_units, units)) #这个需要手动输入一下输入特征数目还有神经元数目
        self.bias = nn.Parameter(torch.randn(units,))            
    def forward(self, X):
        linear = torch.matmul(X, self.weight.data) + self.bias.data  #向前传播其实就是接受输入
        return F.relu(linear)
#创建一个层，这个层可以直接用在sequential之中
linear = MyLinear(5, 3) #五个输入三个神经元

这里可以看到只要重写了forward方法，那么这个类就能变成一个能用来计算的类，甚至是一个层可以单独计算。并且这样子写好以后是可以放在sequential容器中，作为一个统一训练的。

因此，如果我们有多个块的话，也是可以自己去写一个容器，进行组合。

python 复制代码

#创建一个新类
class MySequential(nn.Module): #（）就是py中的继承语法
    def __init__(self, *args):
        super().__init__()
        for idx, module in enumerate(args):
            # 这里，module是Module子类的一个实例。我们把它保存在'Module'类的成员
            # 变量_modules中。_module的类型是OrderedDict
            self._modules[str(idx)] = module

    def forward(self, X):
        # OrderedDict保证了按照成员添加的顺序遍历它们
        for block in self._modules.values():
            X = block(X)
        return X
    

#这个类实现的效果就类似原声的sequential
net = MySequential(net1，net2，net3)
print(net(X))

这个就大概是在拼接块，层的时候，内部所做的底层原理。

当然直接用sequential容器是更加省力气的方法，对吧

2.关于参数如何进行检查

假设现在有一个单独的神经网络

python 复制代码

net = nn.Sequential(nn.Linear(4, 8), nn.ReLU(), nn.Linear(8, 1))

众所周知，这个神经网络是两层（中间的一层是激活函数我们不做讨论）

我们可以通过索引来调用和获取某个层的属性

python 复制代码

#返回结果是这个全链接层的weight和bias，正好对应八个神经元
print(net[0].state_dict())
#检查参数
print(net[2].bias) #还会返回一些具体的属性
print(net[2].bias.data) #单纯的数据

对于block组成的神经网路社区中（我也不知道很多块组在一起应该叫什么了），仍然是一个嵌套的结构，我们可以创建这样一个社区

python 复制代码

#这段代码其实也能看出来，sequential也是一个能容纳block的东西
#     容器 --》 block --》 layer --》 神经元   这三层架构（或者说四层）
def block1():
    return nn.Sequential(nn.Linear(4, 8), nn.ReLU(),
                         nn.Linear(8, 4), nn.ReLU())
def block2():
    net = nn.Sequential()
    for i in range(4):
        # 在这里嵌套
        net.add_module(f'block {i}', block1())
    return net
    
rgnet = nn.Sequential(block2(), nn.Linear(4, 1))

然后我们对这个rgnet进行打印，可以直接看到工作状态

python 复制代码

#这样子打印会展示整个网络的状态
print('检查工作状态',rgnet)

可以很清晰地看到，这样一个嵌套结构

所以比如说我们想要访问第一个社区中，第2个块，中的第一个层中的参数，我们可以直接这样子读取

python 复制代码

rgnet[0][1][0].bias.data

另外如果想要对已经形成的模型做初始化，这里还有一个例子

python 复制代码

def init_normal(m):
    if type(m) == nn.Linear:
        nn.init.normal_(m.weight, mean=0, std=0.01)   #平均值0,标准差为0.01
        nn.init.zeros_(m.bias)                        #偏移直接设置为0
net.apply(init_normal)
print('手动初始化的效果为',net[0].weight.data[0],'手动初始化bias:', net[0].bias.data[0])

函数实现的功能是先检测传进来的是不是正常的线性层，然后分别初始化。

补充一下，apply函数和js里的用法差不多，对内部的每个单元进行遍历，然后做一些操作。

（当然这不是唯一一种方法，自然还有别的）。

3.关于张量的保存和获取

在pytorch中，张量的保存主要有两种形式，第一种是保存数据，用于其他模型的训练

python 复制代码

#===========读写张量===========#
x = torch.arange(4)       #[0,1,2,3]，创建了一个张量
torch.save(x, 'x-file')   #这是保存在x-file这个文件下面的
loaded_x = torch.load('x-file')  #反过来加载
print(loaded_x)                  #输出
#这样子读取列表和读出，也可以使用字典{x:x,Y:y}或者列表[x,y]，反正是变成文件形式了

另一种是保存模型的参数，可以直接套在其他模型上

python 复制代码

#=====读写参数并且保存在内存=====#

class MLP(nn.Module):   #手动创建多层感知机
    def __init__(self):
        super().__init__()
        self.hidden = nn.Linear(20, 256)
        self.output = nn.Linear(256, 10)

    def forward(self, x):
        return self.output(F.relu(self.hidden(x)))

net = MLP()           #构建对象
print('MLP的参数',net.state_dict()) #这里输出一下参数

#保存这个模型的参数
torch.save(net.state_dict(), 'mlp.params')

#然后对一个新模型使用这个参数
clone = MLP()
clone.load_state_dict(torch.load('mlp.params'))  #内置函数加载参数

clone.eval()#设置为评估模式，禁止训练什么的，这应该是module中附带的功能

print('clone的参数',clone.state_dict())

#可以看到参数被完全复制了

但是注意一个问题，如果使用另一个模型初始化自身的时候，要保证两个模型的结构一致