sklearn之二元线性回归

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


# 最小二乘法
def compute_error(k, b, x, y):
    total_error = 0
    for i in range(0, len(x)):
        total_error += (y[i] - (k * x[i] + b)) ** 2
    return total_error / float(len(x)) / 2.0


# 读入数据
Data = genfromtxt(r"data2.csv", delimiter=',')

# 切分数据
X_data = Data[:, :-1]  # 除最后一列所有列
Z_data = Data[:, -1]  # 最后一列

# 创建模型
model = linear_model.LinearRegression()
model.fit(X_data, Z_data)
print("斜率：", model.coef_)
print("截距：", model.intercept_)

ax = plt.figure().add_subplot(111, projection='3d')
ax.scatter(X_data[:, 0], X_data[:, 1], Z_data, c='r', marker='o', s=100)
x0 = X_data[:, 0]
x1 = X_data[:, 1]
x0, x1 = np.meshgrid(x0, x1)  # 生成方阵
z = model.intercept_ + x0 * model.coef_[0] + x1 * model.coef_[1]

# 画3d图
ax.plot_surface(x0, x1, z)
# 设置坐标轴
ax.set_xlabel('Miles')
ax.set_ylabel('Num of Deliveries')
ax.set_zlabel('Time')
plt.show()