您当前的位置: 首页 >  pytorch

不牌不改

暂无认证

  • 0浏览

    0关注

    422博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

Pytorch学习笔记(四)

不牌不改 发布时间:2022-01-20 22:55:43 ,浏览量:0

下面的若干段代码是循循渐进的,而且保证了每段代码只对前一段代码进行最小程度的修改,方便连续阅读,这也是我学习的过程,有错误的地方希望不吝赐教,共同进步。

纯手工实现线性回归

10个样本,每个样本特征数为1,需要训练的参数为一个特征权重和一个偏置,所以总共两个要训练的超参数。

采用批量梯度下降法,即使用全部样本的梯度更新权重和偏置。

实现代码
"""
手工实现 y=wx+b 线性回归(梯度下降法)
"""

import torch
import numpy as np

def generate_dataset(true_w, true_b, num_examples, random_seed=
关注
打赏
1662186765
查看更多评论
立即登录/注册

微信扫码登录

0.0466s