深度学习笔记1——线性模型相关
说明:代码和数据参考于刘老师的视频教程《PyTorch深度学习实践》
1. y=w*x模型
1.1 源代码:
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):
return x * w
def loss(x, y):
y_pred = forward(x)
return (y_pred - y).
原创
2021-03-11 09:18:33 ·
179 阅读 ·
0 评论