使用线性回归推导函数实现线性回归详解编程语言

import numpy as np  # 导入科学技术框架

import matplotlib.pyplot as plt  # 导入画图工具

# 使用线性回归推导函数求解theta
# T:转置
# dot:矩阵乘或数组点积
# np.linalg.inv:逆矩阵
theta_best = np.linalg.inv(x_b.T.dot(x_b)).dot(x_b.T).dot(y)  # 线性回归推倒函数
print(“the best compare analyse result:/n{}”.format(theta_best))

# 创建测试集x=0, 2
x_new = np.array([[0], [2]])
x_new_b = np.c_[(np.ones((2, 1))), x_new]
print(“the test data is:/n{}”.format(x_new_b))

# 进行预测：x=0时y应该等于3，x=2时y应该等于11
y_predict = x_new_b.dot(theta_best)
print(“the predict result is:/n{}”.format(y_predict))

# 可视化展示
plt.plot(x_new, y_predict, “r-“)  # 预测的线性回归直线
plt.plot(x, y, “b.”)  # 生成的数据打点
plt.axis([0, 2, 0, 15])  # x,y轴设定，x:0~2,y:0~15
plt.show()  # 显示

结果：

可视化：