手动实现线性回归
使用y = 5x + 20来构造数据文章来源地址https://www.toymoban.com/news/detail-561776.html
- 准备数据
- 计算预测值
- 计算损失 把参数的梯度置为0 进行反向传播
- 更新参数
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 <
文章来源:https://www.toymoban.com/news/detail-561776.html
到了这里,关于pytorch手动实现一个简单的线性回归的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!