LR为什么要进行归一化

1 归一化为什么能提高梯度下降法求解最优解的速度?

      斯坦福机器学习视频做了很好的解释:https://class.coursera.org/ml-003/lecture/21

      如下图所示,蓝色的圈圈图代表的是两个特征的等高线。其中左图两个特征X1和X2的区间相差非常大,X1区间是[0,2000],X2区间是[1,5],其所形成的等高线非常尖。当使用梯度下降法寻求最优解时,很有可能走“之字型”路线(垂直等高线走),从而导致需要迭代很多次才能收敛;

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
以下是一个可能的PyTorch代码实现: ```python import numpy as np import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import DataLoader, TensorDataset from sklearn.preprocessing import StandardScaler # 生成一个5维随机训练集和测试集 train_data = np.random.rand(100, 5) test_data = np.random.rand(20, 5) # 定义归一化器 scaler = StandardScaler() # 对训练集进行归一化 scaler.fit(train_data) train_data_norm = scaler.transform(train_data) # 将测试集进行归一化 test_data_norm = scaler.transform(test_data) # 将归一化后的数据集还原为原来的维数 train_data_norm = torch.tensor(train_data_norm, dtype=torch.float32) test_data_norm = torch.tensor(test_data_norm, dtype=torch.float32) train_dataset = TensorDataset(train_data_norm) test_dataset = TensorDataset(test_data_norm) train_loader = DataLoader(train_dataset, batch_size=10, shuffle=True) test_loader = DataLoader(test_dataset, batch_size=10, shuffle=False) # 调用归一化器,对验证集进行归一化 val_data = np.random.rand(30, 5) val_data_norm = scaler.transform(val_data) # 将归一化后的验证集还原为原来的维数 val_data_norm = torch.tensor(val_data_norm, dtype=torch.float32) # 定义一个5维的全连接神经网络 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.fc1 = nn.Linear(5, 10) self.fc2 = nn.Linear(10, 1) def forward(self, x): x = self.fc1(x) x = nn.functional.relu(x) x = nn.functional.dropout(x, p=0.5) x = self.fc2(x) return x # 定义损失函数和优化器 net = Net() criterion = nn.MSELoss() optimizer = optim.SGD(net.parameters(), lr=0.01) # 训练模型 for epoch in range(10): running_loss = 0.0 for i, data in enumerate(train_loader, 0): inputs, = data optimizer.zero_grad() outputs = net(inputs) loss = criterion(outputs, torch.ones_like(outputs)) loss.backward() optimizer.step() running_loss += loss.item() print('Epoch %d loss: %.3f' % (epoch + 1, running_loss / len(train_loader))) # 在测试集上进行测试 with torch.no_grad(): for data in test_loader: inputs, = data outputs = net(inputs) print(outputs) # 在验证集上进行测试 val_data_norm = scaler.transform(val_data) val_data_norm = torch.tensor(val_data_norm, dtype=torch.float32) with torch.no_grad(): outputs = net(val_data_norm) print(outputs) ``` 这个代码片段首先生成了一个5维的随机训练集和测试集,使用`StandardScaler`定义了一个归一化器,对训练集进行归一化,将测试集进行归一化,最后将归一化后的数据集还原为原来的维数。 然后,代码片段定义了一个5维的全连接神经网络,并使用PyTorch内置的`MSELoss`作为损失函数,使用`SGD`优化器进行优化。在训练模型时,代码片段使用`DataLoader`对归一化后的数据集进行批处理,将每个批次的数据作为模型的输入,执行前向传播、计算损失、反向传播、参数更新等操作。 在测试集上进行测试时,代码片段使用`test_loader`对归一化后的测试集进行批处理,将每个批次的数据作为模型的输入,执行前向传播,输出模型的预测结果。 在验证集上进行测试时,代码片段先将验证集进行归一化,然后将归一化后的验证集还原为原来的维数,并使用训练集上的归一化器对其进行归一化,最后将归一化后的验证集作为模型的输入,执行前向传播,输出模型的预测结果。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值