用Pytorch实现线性回归

步骤:

numpy:广播机制

 

 

 

输入和输出值必须要是矩阵

模型都需要继承父类nn.module , Super中是继承了父类的构造_init_()

 父类中Linear函数为构造对象,完成运算单元的计算(x*w+b) 

in_features 代表输入x是几维的, out_features 代表输出y是几维的,权重一般放在右侧

bias是指要不要偏置量 ,默认值是ture

 

python语法:args 1,2,4,3; kwargs:x,y;

 

 Model可调用

 size_average是否对总损失求均值,reduce是否对结果进行降维,criterion需要的参数是y和y^;

 优化层:model.parameters() 无论模型中的权重参数有多少都能进算出来,lr学习率;

 训练:做返现传播时记得梯度清零

 model下面的linear,linear下面的权重和偏置

 训练集上收敛,测试集上发散说明模型过拟合

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值