PyTorch 实现 线型回归-谢TS的博客.pdf
需积分: 3 95 浏览量
2023-07-17
15:41:03
上传
评论
收藏 783KB PDF 举报
https://xiets.blog.csdn.net/article/details/131710997
1/5
PyTorch
实
现
线
型回
归
谢
TS
已
于
2023-07-13 21:24:29
修
改
原
⽂
链
接
：
https://xiets.blog.csdn.net/article/details/131710997
版
权
声
明
：
原
创
⽂
章禁
⽌
转载
专
栏
⽬
录
：
PyTorch
专
栏
（
总
⽬
录
）
PyTorch
相
关
⽹
站
：
PyTorch
官
⽹
：
https://pytorch.org/
PyTorch API
：
https://pytorch.org/docs/stable/torch.html
PyTorch
实
现
线
性
回
归
示
例
，
随
机
⽣
成
⼀
组
⼆
维
坐
标
点
数据
集
（
接
近
⼀
元
⼀
次
⽅
程
分
布
），
然
后
使
⽤
线
性
模
型
去
拟
合
数据
集
，
经
过
训
练
计
算
出
线
性
模
型
参
数
。
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import torch
from torch import nn, optim
class LinearModule(nn.Module):
"""
⾃
定
义
线
型
模
型
(
神
经⽹络
模
型
)
"""
def __init__(self, *, in_features: int, out_features: int):
super().__init__()
#
创
建
线
型
模
型
,
指
定
样本
输
⼊
和
输
出
的
特
征
数
量
, nn.Linear
也
是
⼀个
nn.Module
self.linear = nn.Linear(in_features, out_features)
#
可
以
叠
加其
他
Module
组
合
成
复
杂
的
多
层
神
经⽹络
模
型
,
这
⾥
只
使
⽤
⼀个
简
单
的
线
型
模
型
(
也
可
以
不
派
⽣
⼦
类
,
直
接
实
例
化
nn.Linear)
def forward(self, inputs: torch.Tensor) -> torch.Tensor:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
PyTorch
深
度
学
习
/…
专
栏
收
录
该
内
容
9
篇
⽂
章
1000
订
阅
内
容
来
源
：
csdn.net
作
者
昵
称
：
谢
TS
原
⽂
链
接
：
https://xiets.blog.csdn.net/article/details/131710997
作
者
主
⻚
：
https://xiets.blog.csdn.net
资源评论
