bp神经网络回归定义损失函数与优化器Python代码

这里提供使用PyTorch实现BP神经网络回归的代码，包括定义均方误差损失函数和使用SGD和Adam优化器更新参数的代码：

import torch
import torch.nn as nn
import torch.optim as optim

# 定义神经网络模型
class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(10, 5)
        self.fc2 = nn.Linear(5, 1)

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

# 实例化神经网络模型和损失函数
net = Net()
criterion = nn.MSELoss()

# 定义优化器（使用SGD）
learning_rate = 0.01
optimizer = optim.SGD(net.parameters(), lr=learning_rate)

# 定义优化器（使用Adam）
learning_rate = 0.001
optimizer = optim.Adam(net.parameters(), lr=learning_rate)

# 训练模型（以SGD为例）
for epoch in range(num_epochs):
    for i, data in enumerate(train_loader):
        inputs, labels = data
        optimizer.zero_grad()
        outputs = net(inputs)
        loss = criterion(outputs, labels)
        loss.backward()
        optimizer.step()

        if (i+1) % 100 == 0:
            print('Epoch [{}/{}], Step [{}/{}], Loss: {:.4f}'.format(epoch+1, num_epochs, i+1, total_step, loss.item()))

其中，`Net`是定义神经网络模型的类，`criterion`是定义均方误差损失函数的对象，`optimizer`是定义优化器的对象。在训练模型时，使用`optimizer.zero_grad()`清零梯度，计算预测值`outputs`和损失函数`loss`，调用`loss.backward()`计算梯度，最后使用`optimizer.step()`更新参数。