一个最简单的线性模型,w是权重,一般来说会取随机值,然后不断学习直到与预期相同
如此以此取每个值与真实值的差值,即评估误差
即找一个合适的权重w,使得平均误差最小
上面的是针对单个样本的,后面的是对整个train_set的,后面的那个就是MSE(均方误差)
穷举法找出最小的loss区间
import numpy as np
import matplotlib.pyplot as plt
x_data=[1.0,2.0,3.0]#x
y_data=[2.0,4.0,6.0]#ydef forward(x):return x*w
def loss(x,y):y_pred=forward(x)return pow(y_pred-y,2)
w_list=[]
mse_list=[]
for w in np.arange(0.0,4.1,0.1):print('w=',w)l_sum=0for x_val,y_val in zip(x_data,y_data):y_pre_value=forward(x_val)loss_val=loss(x_val,y_val)l_sum+=loss_valprint('\t',x_val,y_val,y_pre_value,loss_val)print('MSE=',l_sum/3)w_list.append(w)mse_list.append(l_sum/3)