河北工业大学刘二大人

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) :
y_pre=xw
return y_pre;
def loss(x,y):
y_pre=forward(x)
loss=(y_pre-y)(y_pre-y)
return loss
w_list=[]
mse_list=[]
for w in np.arange(0.0,4.1,0.1):
print(“w:”,w) ;
l_sum=0;
for x_ral,y_ral in zip (x_data,y_data):
y_ral_pre=forward(x_ral)
loss_ral=loss(x_ral,y_ral)
l_sum+=loss_ral
print("\t",x_ral,y_ral,y_ral_pre,loss_ral)
print(“MSE:”,l_sum/3)
w_list.append(w)
mse_list.append(l_sum)
plt.plot(w_list,mse_list)
plt.xlabel(“w”)
plt.ylabel(“loss”)
plt.show()