PyTorch学习课后作业-刘二大人课程（二）

作业：try to use the model  y=x*w+b,and draw the cost graph

使用了散点图绘制，未使用np.meshgrid()函数

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt


ax3 = plt.axes(projection='3d')

#y=2*x+2
x_data = [1.0,2.0,3.0]
y_data = [4.0,6.0,8.0]

def forward(x):
    return x*w+b

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

w_list = []
b_list = []
mse_list = []

for w in np.arange(0.0,4.0,0.1):
    for b in np.arange(0.0,4.0,0.1):
        l_sum = 0
        for x_val, y_val in zip(x_data, y_data):
            loss_val = loss(x_val,y_val)
            l_sum += loss_val
        w_list.append(w)
        b_list.append(b)
        mse_list.append(l_sum/3)
#散点图绘制
fig = plt.figure()
ax3.scatter(w_list,b_list,mse_list)

plt.show()