pytorch手动实现一个简单的线性回归

手动实现线性回归

使用y = 5x + 20来构造数据

  1. 准备数据
  2. 计算预测值
  3. 计算损失 把参数的梯度置为0 进行反向传播
  4. 更新参数
learning_rate = 0.01

# 造数据
x = torch.rand([50, 1])
"""
x[:5]
tensor([[0.4528],
        [0.3746],
        [0.8981],
        [0.6542],
        [0.7539]])
"""
y_true = x * 5 <

你可能感兴趣的:(python,笔记记录,pytorch,线性回归,人工智能)