第二讲 线性模型
本人目前在学习b站上刘二大人的Pytorch深度学习实践的课程,准备通过写博客的方式记录自己学习过程,在希望能够和大家进行交流学习。
下面是课堂教学代码。
import numpy as np
import matplotlib.pyplot as plt
#给出的模型为y=2x
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)**2
#穷举法
w_list = []
mse_list = []