Pytorch实现线性回归实战
随机初始化一个二维数据集,数据中只包含X轴Y轴坐标,然后使用Pytorch训练一个一元线性回归模型模拟这些数据。
随机生成二维数据并可视化
import numpy as np
import random
import matplotlib.pyplot as plt
x = np.arange(20)
y = np.array([5 * x[i] + random.randint(1, 20) for i in range(len(x))])
print(x, y)
plt.xlabel("X")
plt
原创
2021-11-15 09:22:00 ·
1680 阅读 ·
0 评论