python多种方式求解线性回归

数据

利用numpy生成20个二维的点（方便可视化）

import matplotlib.pyplot as plt
import numpy as np

np.random.seed(10)
xTrain = np.linspace(0, 30, 20)
yTrain = xTrain * 3 + np.random.normal(0, 5, 20)
plt.figure()
plt.scatter(xTrain, yTrain)
plt.show()

结果如图所示：

利用sklearn中的linear_model模块生成线性回归模型

由于linear_model对变量的格式要求为：[[0, 0], [1, 1], [2, 2]]，与之相对应的格式应为[0, 1, 2]。故需对xTrain做如下处理

xTrainSK = []
for i in xTrain:
    xTrainSK.append([i])

利用数据训练一个线性回归模型