sklearn库LinearRegression线性回归实现

 来自HCIA-AI 实验

from sklearn.linear_model import LinearRegression
import matplotlib.pyplot as plt
import numpy as np

x = np.array([121, 125, 131, 141, 152, 161]).reshape(-1, 1)  # reshape(1,-1)转化成1行：reshape(2,-1)转换成两行：reshape(-1,1)转换成1列：reshape(-1,2)转化成两列
y = np.array([300, 350, 425, 405, 496, 517])

lr = LinearRegression()  # 创建对象
lr.fit(x, y)  # 模型在数据上训练
w = lr.coef_
b = lr.intercept_
print(w, b)  # 截距和斜率

plt.scatter(x, y)  # 显示散点
plt.xlabel("area")
plt.ylabel("price")
plt.plot([x[0], x[-1]],[x[0]*w+b, x[-1]*w+b], color="red")  # 画折线 -1才会输出整条直线
plt.show()  # 显示

testX = np.array([[130]])  # 做出预测
print(lr.predict(testX))

[4.98467124] -274.87696651875774
[373.13029447]