import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
x_data=np.array([1.0,2.0,3.0])
y_data=np.array([5.0,8.0,11.0])
w_arr=np.arange(0.0,4.1,0.1)
b_arr=np.arange(0.0,3.1,0.1)
def forward(x):
return x*w+b
def loss(x,y):
y_pred=forward(x)
return np.sum((y_pred-y)**2)
loss_list=[]
for b in b_arr:
for w in w_arr:
y_pre=forward(x_data)
los=loss(x_data,y_data)
loss_list.append(los)
loss_list=np.array(loss_list).reshape(b_arr.size,w_arr.size)
X,Y=np.meshgrid(w_arr,b_arr)
fig=plt.figure()
ax=Axes3D(fig)
ax.plot_surface(X,Y,loss_list)
plt.show()
b_index,w_index=np.where(loss_list==np.min(loss_list))#求二维矩阵最小索引
b_index=int(b_index);w_index=int(w_index)
b_pre=b_arr[b_index]
w_pre=w_arr[w_index]