pytorch学习（一）线性模型

@@老胡

已于 2024-01-19 11:23:31 修改

阅读量568

点赞数 10

CC 4.0 BY-SA版权

分类专栏： python 文章标签： pytorch 学习人工智能

于 2024-01-16 19:42:13 首次发布

本文链接：https://blog.youkuaiyun.com/CodePlayMe/article/details/135633048

文章目录

pytorch是一个基础的python的科学计算库，它有以下特点：

类似于numpy，但是它可以使用GPU

可以用它来定义深度学习模型，可以灵活的进行深度学习模型的训练和使用

线性模型

线性模型的基本形式为： $f(x)=w^Tx+b$ ，线性模型的参数是w和b，它的学习是通过不断减少损失实现的，其损失一般为均方损失

在这里插入图片描述
pytorch代码实现：

# from tkinter import W
import numpy as np
import matplotlib.pyplot as plt

x_data=[1.0,2.0,3.0]
y_data=[2.0,4.0,6.0]

def forward(x):
    return x*w

def loss(x,y):
    y_pred=forward(x)
    return (y_pred-y)*(y_pred-y)

w_list=[]
mse_list=[]

# 对于不同的w，手动模拟学习的过程
for w in np.ara