Pytorch深度学习实践-刘二大人-线性模型作业

import numpy as np
import matplotlib.pyplot as plt

x_data = [1, 2, 3, 4]
y_data = [2, 4, 6, 8]

w_list = np.arange(0.0, 4.1, 0.1)
b_list = np.arange(-2.0, 2.1, 0.1)
w, b = np.meshgrid(w_list, b_list)


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


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


mse_list = []
for x_val, y_val in zip(x_data, y_data):
    l_sum = 0
    y_pre_val = forward(x_val)
    loss_val = loss(x_val, y_val)
    l_sum += loss_val
    print('\t', x_val, y_val, y_pre_val, loss_val)
print('MSE',  l_sum/4)
mse_list.append(l_sum / 4)


fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.plot_surface(w, b, mse_list[0],  ##x,y,z二维矩阵(坐标矩阵xv,yv,zv)
                rstride=1,  ##retride(row)指定行的跨度
                cstride=1,  ##retride(column)指定列的跨度
                cmap='rainbow')  ##设置颜色映射
plt.show()

实验结果为:

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值