pytorch实现回归算法

二、利用 PyTorch 实现简单的线性回归算法

2.1 准备数据

下面使用 PyTorch 实现一个简单的线性回归算法。

线性回归是机器学习中最基础和简单的算法，你可以将它视为深度学习界的 HelloWorld。如果不了解线性回归，你可以简单的理解为：训练一条直线，让这条直线拟合一些数据点的趋势。

import torch

# 导入自动梯度运算包，主要用Variable这个类

from torch.autograd import Variable

#

import matplotlib.pyplot as plt

# 生成100个0---100间的数

x = Variable(torch.linspace(0, 100, 100).type(torch.FloatTensor))

# 生成100个正态分布随机数，均值为0，方差为10

rand = Variable(torch.randn(100)) * 10

y = x + rand

# 将Variable转换为数组，绘图

plt.plot(x.data.numpy(), y.data.numpy(),'o')

plt.xlabel('X')