多层感知机(MLP)的设计与实现
一、概念1. mlp多层感知机（multilayer perceptron）缩写为mlp，也称作前馈神经网络（feedforward neural network）。 它是一种基于神经网络的机器学习模型，通过多层非线性变换对输入数据进行高级别的抽象和分类。
与单层感知机相比，mlp有多个隐藏层，每个隐藏层由多个神经元组成，每个神经元通过对上一层的输入进行加权和处理，再通过激活函数进行非线性映射。
mlp的输出层通常是一个 softmax 层，用于多分类任务，或者是一个 sigmoid 层，用于二分类任务。
由于它的强大表达能力和灵活性，mlp被广泛应用于各种机器学习任务中。
2. 前向传播由于有多个层，参数需要在这些层之间传递。 首先需要实现的就是参数的前向传播，计算过程如下：
将输入数据传递给第一个隐藏层的神经元；对于每个神经元，计算其加权和，即将输入与对应的权重相乘并求和，再加上偏置项；将加权和输入到激活函数中，得到激活值，作为该神经元的输出；将每个神经元的输出传递到下一层的神经元，直至输出层。在这个过程中，数据和权重是前向传播的主要传播内容。
3. 反向传播利用链式法则对网络中的参数进行梯度更新。 在训练神经网络时，通常需要定义一个损失函数（loss function），用于评估模型预测结果与真实标签之间的差距。 反向传播算法的目标就是最小化这个损失函数。
在反向传播过程中，算法首先计算损失函数对最后一层的输出的梯度，然后根据链式法则逐层向前计算各层的梯度，并利用梯度下降法更新网络中的参数。 具体地，算法会先将损失函数对输出的梯度传回网络最后一层，然后依次向前计算各层的梯度。 在计算梯度的过程中，算法会利用反向传播公式来计算当前层的梯度，然后将这个梯度传递到前一层。 在更新网络参数时，算法会根据计算出的梯度和学习率来更新网络中的权重和偏置。
通过不断地反复迭代前向传播和反向传播两个阶段，可以不断地更新网络中的参数，从而逐渐提高模型的性能。
下面是一个python计算反向传播的示例：
for epoch in range(num_epochs): for x, y_true in zip(x_train, y_train): # 前向传播 hidden_layer = np.maximum(0, np.dot(x, self.weights1) + self.bias1) # relu激活函数 y_pred = np.dot(hidden_layer, self.weights2) + self.bias2 # 计算损失和梯度，使用均方误差作为损失函数(mean squared error,mse) # 对于每一个样本，模型预测出来的输出与实际输出之间的差异会被平方， # 然后对所有样本的平方差进行求和并除以样本数，即可得到mse作为模型的损失函数。 loss = np.square(y_true - y_pred).sum() # 下面复杂的方法用来实现反向传播 # 计算损失函数关于预测输出的导数 d_loss_pred = -2.0 * (y_true - y_pred) # 计算输出层的梯度， d_weights2 = np.dot(hidden_layer.reshape(-1, 1), d_loss_pred.reshape(1, -1)) # 计算输出层偏置的梯度，其值等于输出误差 d_bias2 = d_loss_pred # 计算隐藏层误差，其中 self.weights2.t 代表输出层权重的转置， # 计算得到的结果是一个行向量，代表每个隐藏层节点的误差。 d_hidden = np.dot(d_loss_pred, self.weights2.t) # 将隐藏层误差中小于等于 0 的部分置为 0，相当于计算 relu 激活函数的导数， # 这是因为 relu 函数在小于等于 0 的部分导数为 0 d_hidden[hidden_layer <= 0] = 0 # relu激活函数的导数 # 计算隐藏层权重的梯度， # 其中 x.reshape(-1, 1) 代表将输入变为列向量， # d_hidden.reshape(1, -1) 代表将隐藏层误差变为行向量， # 两者的点积得到的是一个矩阵， # 这个矩阵的行表示输入的维度（也就是输入节点的个数）， # 列表示输出的维度（也就是隐藏层节点的个数），表示每个输入和每个隐藏层节点的权重梯度。 d_weights1 = np.dot(x.reshape(-1, 1), d_hidden.reshape(1, -1)) # 计算隐藏层偏置的梯度，其值等于隐藏层误差。 d_bias1 = d_hidden # 更新权重和偏置 self.weights2 -= learning_rate * d_weights2 self.bias2 -= learning_rate * d_bias2 self.weights1 -= learning_rate * d_weights1 self.bias1 -= learning_rate * d_bias14. 评估模式与训练模式在训练阶段，需要调用 model.train() 来指定模型为训练模式，以便启用训练中需要的特性，如 dropout 和 batch normalization 等。如果在训练过程中需要评估模型的性能，可以在评估前调用 model.eval()，以确保模型处于评估模式，而不是训练模式。在测试阶段，需要调用 model.eval() 来指定模型为评估模式，以便禁用 dropout 和 batch normalization 等特性，以及启用测试中需要的特性，如在计算中间层的输出等。在预测阶段，需要调用 model.eval() 来指定模型为预测模式，以便禁用 dropout 和 batch normalization 等特性，并且只计算模型的前向传播，以生成模型的输出，而不更新模型的权重。二、模型定义1. 加载数据集本示例使用 mnist 数据集。
import torchfrom torchvision import datasets, transformsfrom torch.autograd import variableimport timeimport matplotlib.pyplot as plt# 定义totensor和normalize的transformto_tensor = transforms.totensor()normalize = transforms.normalize((0.5,), (0.5,))# 定义compose的transformtransform = transforms.compose([ to_tensor, # 转换为张量 normalize # 标准化])# 下载数据集data_train = datasets.mnist(root=..//data//, transform=transform, train=true, download=true)data_test = datasets.mnist(root=..//data//, transform=transform, train=false, download=true)# 装载数据data_loader_train = torch.utils.data.dataloader(dataset=data_train, batch_size=64, shuffle=true)data_loader_test = torch.utils.data.dataloader(dataset=data_test, batch_size=64, shuffle=true)2. 定义mlp层下面定义一个有三个层的mlp。
对于这个mlp，它接收一个num_i的输入，输出为num_o的预测值。 隐藏层有2层，每层大小为num_h。
层的定义如下：
self.linear1：输入层到第一层隐藏层的线性转换，其中num_i为输入的特征数，num_h为第一层隐藏层的特征数。self.relu：第一层隐藏层的激活函数，采用relu。self.linear2：第一层隐藏层到第二层隐藏层的线性转换，其中num_h为第一层隐藏层的特征数，num_h为第二层隐藏层的特征数。self.relu2：第二层隐藏层的激活函数，采用relu。self.linear3：第二层隐藏层到输出层的线性转换，其中num_h为第二层隐藏层的特征数，num_o为输出的特征数（或者说类别数）。class mlp(torch.nn.module): def __init__(self, num_i, num_h, num_o): super(mlp, self).__init__() self.linear1 = torch.nn.linear(num_i, num_h) self.relu = torch.nn.relu() self.linear2 = torch.nn.linear(num_h, num_h) # 2个隐层 self.relu2 = torch.nn.relu() self.linear3 = torch.nn.linear(num_h, num_o)3. 前向传播def forward(self, x): x = self.linear1(x) x = self.relu(x) x = self.linear2(x) x = self.relu2(x) x = self.linear3(x) return x在前向传播时，输入x先通过第一层的线性转换，然后经过第一层隐藏层的激活函数，
再通过第二层的线性转换，再经过第二层隐藏层的激活函数，
最后输出预测值。
4. 优化器本文将使用pytorch的优化器工具用于反向传播 。
优化器(optimizer)是一个用于更新模型参数的工具，根据训练集的损失函数(loss function)和反向传播算法(backpropagation algorithm)计算梯度，并使用梯度下降算法(gradient descent algorithm)更新模型参数，以最小化损失函数的值。
pytorch提供了许多常用的优化器，如随机梯度下降法(sgd)、adam、adagrad、rmsprop等。
这些优化器使用不同的更新策略，根据不同的训练任务和数据集选择合适的优化器可以提高训练效率和性能。
5. 反向传播本文使用和pytorch优化器的一个实例： torch.optim.adam()，它使用反向传播算法计算梯度并更新模型的权重，从而调整模型参数以最小化损失函数。
三、训练def train(model): cost = torch.nn.crossentropyloss() optimizer = torch.optim.adam(model.parameters()) # 设置迭代次数 epochs = 5 for epoch in range(epochs): sum_loss = 0 train_correct = 0 for data in data_loader_train: # 获取数据和标签 inputs, labels = data # inputs 维度：[64,1,28,28] # 将输入数据展平为一维向量 inputs = torch.flatten(inputs, start_dim=1) # 展平数据，转化为[64,784] # 计算输出 outputs = model(inputs) # 将梯度清零 optimizer.zero_grad() # 计算损失函数 loss = cost(outputs, labels) # 反向传播计算梯度 loss.backward() # 使用优化器更新模型参数 optimizer.step() # 返回 outputs 张量每行中的最大值和对应的索引，1表示从行维度中找到最大值 _, id = torch.max(outputs.data, 1) # 将每个小批次的损失值 loss 累加，用于最后计算平均损失 sum_loss += loss.data # 计算每个小批次正确分类的图像数量 train_correct += torch.sum(id == labels.data) print('[%d/%d] loss:%.3f, correct:%.3f%%, time:%s' % (epoch + 1, epochs, sum_loss / len(data_loader_train), 100 * train_correct / len(data_train), time.strftime('%y-%m-%d %h:%m:%s', time.localtime()))) model.eval()四、测试与预测# 测试模型def test(model, test_loader): test_correct = 0 for data in test_loader: inputs, lables = data inputs, lables = variable(inputs).cpu(), variable(lables).cpu() inputs = torch.flatten(inputs, start_dim=1) # 展并数据 outputs = model(inputs) _, id = torch.max(outputs.data, 1) test_correct += torch.sum(id == lables.data) print(f'accuracy on test set: {100 * test_correct / len(data_test):.3f}%')五、预测在神经网络模型的推断（inference）阶段中，我们不需要进行反向传播，也不需要计算梯度，使用 with torch.no_grad(): 上下文管理器可以有效地减少内存消耗和计算时间
def predict(model, data): model.eval() with torch.no_grad(): output = model(data) pred = output.data.max(1, keepdim=true)[1] return pred1234567output是模型在给定输入数据后的输出结果，每一行对应一个输入数据样本，每一列对应一个可能的输出类别。output.data提取出了output中的数据部分，然后使用max()函数沿着第1个维度(即列)找到每一行中最大的值以及对应的索引。keepdim=true参数使得输出结果保持和输入数据output相同的维度。 因此，pred包含每个输入样本的预测类别，是一个包含预测标签索引的一维张量六、完整代码import torchfrom torchvision import datasets, transformsfrom torch.autograd import variableimport timeimport matplotlib.pyplot as plt# 定义totensor和normalize的transformto_tensor = transforms.totensor()normalize = transforms.normalize((0.5,), (0.5,))# 定义compose的transformtransform = transforms.compose([ to_tensor, # 转换为张量 normalize # 标准化])# 下载数据集data_train = datasets.mnist(root=..//data//, transform=transform, train=true, download=true)data_test = datasets.mnist(root=..//data//, transform=transform, train=false, download=true)# 装载数据data_loader_train = torch.utils.data.dataloader(dataset=data_train, batch_size=64, shuffle=true)data_loader_test = torch.utils.data.dataloader(dataset=data_test, batch_size=64, shuffle=true)class mlp(torch.nn.module): def __init__(self, num_i, num_h, num_o): super(mlp, self).__init__() self.linear1 = torch.nn.linear(num_i, num_h) self.relu = torch.nn.relu() self.linear2 = torch.nn.linear(num_h, num_h) # 2个隐层 self.relu2 = torch.nn.relu() self.linear3 = torch.nn.linear(num_h, num_o) def forward(self, x): x = self.linear1(x) x = self.relu(x) x = self.linear2(x) x = self.relu2(x) x = self.linear3(x) return xdef train(model): # 损失函数，它将网络的输出和目标标签进行比较，并计算它们之间的差异。在训练期间，我们尝试最小化损失函数，以使输出与标签更接近 cost = torch.nn.crossentropyloss() # 优化器的一个实例，用于调整模型参数以最小化损失函数。 # 使用反向传播算法计算梯度并更新模型的权重。在这里，我们使用adam优化器来优化模型。model.parameters()提供了要优化的参数。 optimizer = torch.optim.adam(model.parameters()) # 设置迭代次数 epochs = 2 for epoch in range(epochs): sum_loss = 0 train_correct = 0 for data in data_loader_train: inputs, labels = data # inputs 维度：[64,1,28,28] # print(inputs.shape) inputs = torch.flatten(inputs, start_dim=1) # 展平数据，转化为[64,784] # print(inputs.shape) outputs = model(inputs) optimizer.zero_grad() loss = cost(outputs, labels) loss.backward() optimizer.step() _, id = torch.max(outputs.data, 1) sum_loss += loss.data train_correct += torch.sum(id == labels.data) print('[%d/%d] loss:%.3f, correct:%.3f%%, time:%s' % (epoch + 1, epochs, sum_loss / len(data_loader_train), 100 * train_correct / len(data_train), time.strftime('%y-%m-%d %h:%m:%s', time.localtime()))) model.train()# 测试模型def test(model, test_loader): model.eval() test_correct = 0 for data in test_loader: inputs, lables = data inputs, lables = variable(inputs).cpu(), variable(lables).cpu() inputs = torch.flatten(inputs, start_dim=1) # 展并数据 outputs = model(inputs) _, id = torch.max(outputs.data, 1) test_correct += torch.sum(id == lables.data) print(f'accuracy on test set: {100 * test_correct / len(data_test):.3f}%')# 预测模型def predict(model, data): model.eval() with torch.no_grad(): output = model(data) pred = output.data.max(1, keepdim=true)[1] return prednum_i = 28 * 28 # 输入层节点数num_h = 100 # 隐含层节点数num_o = 10 # 输出层节点数batch_size = 64model = mlp(num_i, num_h, num_o)train(model)test(model, data_loader_test)# 预测图片，这里取测试集前10张图片for i in range(10): # 获取测试数据中的第一张图片 test_image = data_test[i][0] # 展平图片 test_image = test_image.flatten() # 增加一维作为 batch 维度 test_image = test_image.unsqueeze(0) # 显示图片 plt.imshow(test_image.view(28, 28), cmap='gray') plt.show() pred = predict(model, test_image) print('prediction:', pred.item())
智信通|全球领先的技术服务商为汽车金融带来全新解决方案
一键连接，轻松实现数据传输，没有什么比这款DTU更合适
华光光电推出医疗美容用100W激光器
LED植物照明产业发展三大趋势
2017年与2019年的比特币价格对比分析
多层感知机(MLP)的设计与实现
获得可靠的传感器数据：电子、光学和机械设计之间的相互作用
更新IoT设备OTA的方法
全球最快电池摩托车
Linux驱动高精度定时器hrtimer
SiCMOSFET如何实现降低功率转换过程中能量损耗
中性线零线和地线接错有什么后果
Altium designer生成PCB图的方法
谷歌推出视频智能API 自动识别物体
可控硅整流器件
TE开发出用于汽车线束的REM0.64端子
新加坡央行计划利用区块链来提高国际支付的效率
电动牙刷气如何进行密性防水测试
厉害了！通用电气将AR增强现实技术用于机械组装
锐龙7 2700对比 i5-9600KF测评
与单层感知机相比，mlp有多个隐藏层，每个隐藏层由多个神经元组成，每个神经元通过对上一层的输入进行加权和处理，再通过激活函数进行非线性映射。
mlp的输出层通常是一个 softmax 层，用于多分类任务，或者是一个 sigmoid 层，用于二分类任务。
由于它的强大表达能力和灵活性，mlp被广泛应用于各种机器学习任务中。
2. 前向传播由于有多个层，参数需要在这些层之间传递。 首先需要实现的就是参数的前向传播，计算过程如下：
将输入数据传递给第一个隐藏层的神经元；对于每个神经元，计算其加权和，即将输入与对应的权重相乘并求和，再加上偏置项；将加权和输入到激活函数中，得到激活值，作为该神经元的输出；将每个神经元的输出传递到下一层的神经元，直至输出层。在这个过程中，数据和权重是前向传播的主要传播内容。
3. 反向传播利用链式法则对网络中的参数进行梯度更新。 在训练神经网络时，通常需要定义一个损失函数（loss function），用于评估模型预测结果与真实标签之间的差距。 反向传播算法的目标就是最小化这个损失函数。
在反向传播过程中，算法首先计算损失函数对最后一层的输出的梯度，然后根据链式法则逐层向前计算各层的梯度，并利用梯度下降法更新网络中的参数。 具体地，算法会先将损失函数对输出的梯度传回网络最后一层，然后依次向前计算各层的梯度。 在计算梯度的过程中，算法会利用反向传播公式来计算当前层的梯度，然后将这个梯度传递到前一层。 在更新网络参数时，算法会根据计算出的梯度和学习率来更新网络中的权重和偏置。
通过不断地反复迭代前向传播和反向传播两个阶段，可以不断地更新网络中的参数，从而逐渐提高模型的性能。
下面是一个python计算反向传播的示例：
for epoch in range(num_epochs): for x, y_true in zip(x_train, y_train): # 前向传播 hidden_layer = np.maximum(0, np.dot(x, self.weights1) + self.bias1) # relu激活函数 y_pred = np.dot(hidden_layer, self.weights2) + self.bias2 # 计算损失和梯度，使用均方误差作为损失函数(mean squared error,mse) # 对于每一个样本，模型预测出来的输出与实际输出之间的差异会被平方， # 然后对所有样本的平方差进行求和并除以样本数，即可得到mse作为模型的损失函数。 loss = np.square(y_true - y_pred).sum() # 下面复杂的方法用来实现反向传播 # 计算损失函数关于预测输出的导数 d_loss_pred = -2.0 * (y_true - y_pred) # 计算输出层的梯度， d_weights2 = np.dot(hidden_layer.reshape(-1, 1), d_loss_pred.reshape(1, -1)) # 计算输出层偏置的梯度，其值等于输出误差 d_bias2 = d_loss_pred # 计算隐藏层误差，其中 self.weights2.t 代表输出层权重的转置， # 计算得到的结果是一个行向量，代表每个隐藏层节点的误差。 d_hidden = np.dot(d_loss_pred, self.weights2.t) # 将隐藏层误差中小于等于 0 的部分置为 0，相当于计算 relu 激活函数的导数， # 这是因为 relu 函数在小于等于 0 的部分导数为 0 d_hidden[hidden_layer <= 0] = 0 # relu激活函数的导数 # 计算隐藏层权重的梯度， # 其中 x.reshape(-1, 1) 代表将输入变为列向量， # d_hidden.reshape(1, -1) 代表将隐藏层误差变为行向量， # 两者的点积得到的是一个矩阵， # 这个矩阵的行表示输入的维度（也就是输入节点的个数）， # 列表示输出的维度（也就是隐藏层节点的个数），表示每个输入和每个隐藏层节点的权重梯度。 d_weights1 = np.dot(x.reshape(-1, 1), d_hidden.reshape(1, -1)) # 计算隐藏层偏置的梯度，其值等于隐藏层误差。 d_bias1 = d_hidden # 更新权重和偏置 self.weights2 -= learning_rate * d_weights2 self.bias2 -= learning_rate * d_bias2 self.weights1 -= learning_rate * d_weights1 self.bias1 -= learning_rate * d_bias14. 评估模式与训练模式在训练阶段，需要调用 model.train() 来指定模型为训练模式，以便启用训练中需要的特性，如 dropout 和 batch normalization 等。如果在训练过程中需要评估模型的性能，可以在评估前调用 model.eval()，以确保模型处于评估模式，而不是训练模式。在测试阶段，需要调用 model.eval() 来指定模型为评估模式，以便禁用 dropout 和 batch normalization 等特性，以及启用测试中需要的特性，如在计算中间层的输出等。在预测阶段，需要调用 model.eval() 来指定模型为预测模式，以便禁用 dropout 和 batch normalization 等特性，并且只计算模型的前向传播，以生成模型的输出，而不更新模型的权重。二、模型定义1. 加载数据集本示例使用 mnist 数据集。
import torchfrom torchvision import datasets, transformsfrom torch.autograd import variableimport timeimport matplotlib.pyplot as plt# 定义totensor和normalize的transformto_tensor = transforms.totensor()normalize = transforms.normalize((0.5,), (0.5,))# 定义compose的transformtransform = transforms.compose([ to_tensor, # 转换为张量 normalize # 标准化])# 下载数据集data_train = datasets.mnist(root=..//data//, transform=transform, train=true, download=true)data_test = datasets.mnist(root=..//data//, transform=transform, train=false, download=true)# 装载数据data_loader_train = torch.utils.data.dataloader(dataset=data_train, batch_size=64, shuffle=true)data_loader_test = torch.utils.data.dataloader(dataset=data_test, batch_size=64, shuffle=true)2. 定义mlp层下面定义一个有三个层的mlp。
对于这个mlp，它接收一个num_i的输入，输出为num_o的预测值。 隐藏层有2层，每层大小为num_h。
层的定义如下：
self.linear1：输入层到第一层隐藏层的线性转换，其中num_i为输入的特征数，num_h为第一层隐藏层的特征数。self.relu：第一层隐藏层的激活函数，采用relu。self.linear2：第一层隐藏层到第二层隐藏层的线性转换，其中num_h为第一层隐藏层的特征数，num_h为第二层隐藏层的特征数。self.relu2：第二层隐藏层的激活函数，采用relu。self.linear3：第二层隐藏层到输出层的线性转换，其中num_h为第二层隐藏层的特征数，num_o为输出的特征数（或者说类别数）。class mlp(torch.nn.module): def __init__(self, num_i, num_h, num_o): super(mlp, self).__init__() self.linear1 = torch.nn.linear(num_i, num_h) self.relu = torch.nn.relu() self.linear2 = torch.nn.linear(num_h, num_h) # 2个隐层 self.relu2 = torch.nn.relu() self.linear3 = torch.nn.linear(num_h, num_o)3. 前向传播def forward(self, x): x = self.linear1(x) x = self.relu(x) x = self.linear2(x) x = self.relu2(x) x = self.linear3(x) return x在前向传播时，输入x先通过第一层的线性转换，然后经过第一层隐藏层的激活函数，
再通过第二层的线性转换，再经过第二层隐藏层的激活函数，
最后输出预测值。
4. 优化器本文将使用pytorch的优化器工具用于反向传播 。
优化器(optimizer)是一个用于更新模型参数的工具，根据训练集的损失函数(loss function)和反向传播算法(backpropagation algorithm)计算梯度，并使用梯度下降算法(gradient descent algorithm)更新模型参数，以最小化损失函数的值。
pytorch提供了许多常用的优化器，如随机梯度下降法(sgd)、adam、adagrad、rmsprop等。
这些优化器使用不同的更新策略，根据不同的训练任务和数据集选择合适的优化器可以提高训练效率和性能。
5. 反向传播本文使用和pytorch优化器的一个实例： torch.optim.adam()，它使用反向传播算法计算梯度并更新模型的权重，从而调整模型参数以最小化损失函数。
三、训练def train(model): cost = torch.nn.crossentropyloss() optimizer = torch.optim.adam(model.parameters()) # 设置迭代次数 epochs = 5 for epoch in range(epochs): sum_loss = 0 train_correct = 0 for data in data_loader_train: # 获取数据和标签 inputs, labels = data # inputs 维度：[64,1,28,28] # 将输入数据展平为一维向量 inputs = torch.flatten(inputs, start_dim=1) # 展平数据，转化为[64,784] # 计算输出 outputs = model(inputs) # 将梯度清零 optimizer.zero_grad() # 计算损失函数 loss = cost(outputs, labels) # 反向传播计算梯度 loss.backward() # 使用优化器更新模型参数 optimizer.step() # 返回 outputs 张量每行中的最大值和对应的索引，1表示从行维度中找到最大值 _, id = torch.max(outputs.data, 1) # 将每个小批次的损失值 loss 累加，用于最后计算平均损失 sum_loss += loss.data # 计算每个小批次正确分类的图像数量 train_correct += torch.sum(id == labels.data) print('[%d/%d] loss:%.3f, correct:%.3f%%, time:%s' % (epoch + 1, epochs, sum_loss / len(data_loader_train), 100 * train_correct / len(data_train), time.strftime('%y-%m-%d %h:%m:%s', time.localtime()))) model.eval()四、测试与预测# 测试模型def test(model, test_loader): test_correct = 0 for data in test_loader: inputs, lables = data inputs, lables = variable(inputs).cpu(), variable(lables).cpu() inputs = torch.flatten(inputs, start_dim=1) # 展并数据 outputs = model(inputs) _, id = torch.max(outputs.data, 1) test_correct += torch.sum(id == lables.data) print(f'accuracy on test set: {100 * test_correct / len(data_test):.3f}%')五、预测在神经网络模型的推断（inference）阶段中，我们不需要进行反向传播，也不需要计算梯度，使用 with torch.no_grad(): 上下文管理器可以有效地减少内存消耗和计算时间
def predict(model, data): model.eval() with torch.no_grad(): output = model(data) pred = output.data.max(1, keepdim=true)[1] return pred1234567output是模型在给定输入数据后的输出结果，每一行对应一个输入数据样本，每一列对应一个可能的输出类别。output.data提取出了output中的数据部分，然后使用max()函数沿着第1个维度(即列)找到每一行中最大的值以及对应的索引。keepdim=true参数使得输出结果保持和输入数据output相同的维度。 因此，pred包含每个输入样本的预测类别，是一个包含预测标签索引的一维张量六、完整代码import torchfrom torchvision import datasets, transformsfrom torch.autograd import variableimport timeimport matplotlib.pyplot as plt# 定义totensor和normalize的transformto_tensor = transforms.totensor()normalize = transforms.normalize((0.5,), (0.5,))# 定义compose的transformtransform = transforms.compose([ to_tensor, # 转换为张量 normalize # 标准化])# 下载数据集data_train = datasets.mnist(root=..//data//, transform=transform, train=true, download=true)data_test = datasets.mnist(root=..//data//, transform=transform, train=false, download=true)# 装载数据data_loader_train = torch.utils.data.dataloader(dataset=data_train, batch_size=64, shuffle=true)data_loader_test = torch.utils.data.dataloader(dataset=data_test, batch_size=64, shuffle=true)class mlp(torch.nn.module): def __init__(self, num_i, num_h, num_o): super(mlp, self).__init__() self.linear1 = torch.nn.linear(num_i, num_h) self.relu = torch.nn.relu() self.linear2 = torch.nn.linear(num_h, num_h) # 2个隐层 self.relu2 = torch.nn.relu() self.linear3 = torch.nn.linear(num_h, num_o) def forward(self, x): x = self.linear1(x) x = self.relu(x) x = self.linear2(x) x = self.relu2(x) x = self.linear3(x) return xdef train(model): # 损失函数，它将网络的输出和目标标签进行比较，并计算它们之间的差异。在训练期间，我们尝试最小化损失函数，以使输出与标签更接近 cost = torch.nn.crossentropyloss() # 优化器的一个实例，用于调整模型参数以最小化损失函数。 # 使用反向传播算法计算梯度并更新模型的权重。在这里，我们使用adam优化器来优化模型。model.parameters()提供了要优化的参数。 optimizer = torch.optim.adam(model.parameters()) # 设置迭代次数 epochs = 2 for epoch in range(epochs): sum_loss = 0 train_correct = 0 for data in data_loader_train: inputs, labels = data # inputs 维度：[64,1,28,28] # print(inputs.shape) inputs = torch.flatten(inputs, start_dim=1) # 展平数据，转化为[64,784] # print(inputs.shape) outputs = model(inputs) optimizer.zero_grad() loss = cost(outputs, labels) loss.backward() optimizer.step() _, id = torch.max(outputs.data, 1) sum_loss += loss.data train_correct += torch.sum(id == labels.data) print('[%d/%d] loss:%.3f, correct:%.3f%%, time:%s' % (epoch + 1, epochs, sum_loss / len(data_loader_train), 100 * train_correct / len(data_train), time.strftime('%y-%m-%d %h:%m:%s', time.localtime()))) model.train()# 测试模型def test(model, test_loader): model.eval() test_correct = 0 for data in test_loader: inputs, lables = data inputs, lables = variable(inputs).cpu(), variable(lables).cpu() inputs = torch.flatten(inputs, start_dim=1) # 展并数据 outputs = model(inputs) _, id = torch.max(outputs.data, 1) test_correct += torch.sum(id == lables.data) print(f'accuracy on test set: {100 * test_correct / len(data_test):.3f}%')# 预测模型def predict(model, data): model.eval() with torch.no_grad(): output = model(data) pred = output.data.max(1, keepdim=true)[1] return prednum_i = 28 * 28 # 输入层节点数num_h = 100 # 隐含层节点数num_o = 10 # 输出层节点数batch_size = 64model = mlp(num_i, num_h, num_o)train(model)test(model, data_loader_test)# 预测图片，这里取测试集前10张图片for i in range(10): # 获取测试数据中的第一张图片 test_image = data_test[i][0] # 展平图片 test_image = test_image.flatten() # 增加一维作为 batch 维度 test_image = test_image.unsqueeze(0) # 显示图片 plt.imshow(test_image.view(28, 28), cmap='gray') plt.show() pred = predict(model, test_image) print('prediction:', pred.item())
智信通|全球领先的技术服务商为汽车金融带来全新解决方案
一键连接，轻松实现数据传输，没有什么比这款DTU更合适
华光光电推出医疗美容用100W激光器
LED植物照明产业发展三大趋势
2017年与2019年的比特币价格对比分析
多层感知机(MLP)的设计与实现
获得可靠的传感器数据：电子、光学和机械设计之间的相互作用
更新IoT设备OTA的方法
全球最快电池摩托车
Linux驱动高精度定时器hrtimer
SiCMOSFET如何实现降低功率转换过程中能量损耗
中性线零线和地线接错有什么后果
Altium designer生成PCB图的方法
谷歌推出视频智能API 自动识别物体
可控硅整流器件
TE开发出用于汽车线束的REM0.64端子
新加坡央行计划利用区块链来提高国际支付的效率
电动牙刷气如何进行密性防水测试
厉害了！通用电气将AR增强现实技术用于机械组装
锐龙7 2700对比 i5-9600KF测评