线性回归-案例1

最新推荐文章于 2022-04-06 13:08:08 发布

LL_QQ63

最新推荐文章于 2022-04-06 13:08:08 发布

阅读量156

点赞数

分类专栏：机器算法

本文链接：https://blog.csdn.net/LL_QQ63/article/details/103777931

版权

机器算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏



from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.preprocessing import MinMaxScaler

# 线性回归:正规方程
# 1.获取数据
data = load_boston()

# 2.数据集划分
x_train, x_test, y_train, y_test = train_test_split(data.data, data.target, random_state=22)

# 3.1数据标准化
#transfer = StandardScaler()
#x_train = transfer.fit_transform(x_train)
#x_test = transfer.fit_transform(x_test)

#3.2数据归一化
scaler = MinMaxScaler() 
x_train = scaler.fit_transform(x_train)
x_test = scaler.fit_transform(x_test)

# 4.机器学习-线性回归(正规方程)
estimator = LinearRegression()
estimator.fit(x_train, y_train)

# 5.模型评估
# 5.1 获取系数等值
y_predict = estimator.predict(x_test)
print("预测值为: ", y_predict)
print("模型中的系数为: ", estimator.coef_)
print("模型中的偏置为: ", estimator.intercept_)

# 5.2 评价
# 回归性能评估:均方误差(Mean Squared Error)MSE 评价机制
error = mean_squared_error(y_test, y_predict)
print("误差为: ", error)
#通过数据标准化运行，误差为： 20.0621939903598
#通过数据归一化运行，误差为： 31.803119016104684