python sklearn 线性回归模型 模型评估

python sklearn 线性回归模型

文件下载
链接: https://pan.baidu.com/s/1UQydHMXQQAKiYdBc6VFbOA 提取码: wfde

"""
    使用 sklearn 线性模型 线性回归
"""
import numpy as np
import matplotlib.pyplot as mp
import sklearn.linear_model as lm
import sklearn.metrics as sm

# 读取文本
x, y = np.loadtxt("./single.txt", delimiter=",", unpack=True)

# 整理输入集
x = x.reshape(-1, 1)

# 绘制散点
mp.figure("Linear Regression", facecolor="lightgray")
mp.title("Linear Regression", fontsize=16)
mp.grid(linestyle=":")
mp.scatter(x, y, s=70, color="dodgerblue", label="samples")

# 构建线性回归模型
model = lm.LinearRegression()
model.fit(x, y)
# 针对训练数据 得到结果
prd_y = model.predict(x)
mp.plot(x, prd_y, color="orangered", label="Predict")

mp.legend()
mp.tight_layout()

# 评估当前模型
print("平均绝对值误差:", sm.mean_absolute_error(y, prd_y))
print("平均平方误差:", sm.mean_squared_error(y, prd_y))
print("中位绝对值误差:", sm.median_absolute_error(y, prd_y))
print("R2得分:", sm.r2_score(y, prd_y))

mp.show()