sklearn基础--『监督学习』之岭回归

岭回归 (Ridge Regression)是一种用于处理共线性数据 的线性回归改进方法。

和上一篇用基于最小二乘法的线性回归相比,它通过放弃最小二乘的无偏性,

以损失部分信息、降低精度为代价来获得更实际和可靠性更强的回归系数。

1. 概述

岭回归的模型对于存在大量相关特征(这些特征之间存在很高的相关性)的数据时效果远好于基于最小二乘法的线性模型。

原因就是它通过给系数的大小增加一个约束条件(即L2正则化项 ),来防止模型过度拟合训练数据。

损失函数一般定义为: <math xmlns="http://www.w3.org/1998/Math/MathML"> L ( w ) = ( y − w X ) 2 + λ ∥ w ∥ 2 L(w) = (y-wX)^2+\lambda\parallel w\parallel_2 </math>L(w)=(y−wX)2+λ∥w∥2

其中 <math xmlns="http://www.w3.org/1998/Math/MathML"> λ ∥ w ∥ 2 = λ ∑ i = 1 n w i 2 \lambda\parallel w\parallel_2 = \lambda\sum_{i=1}^{n}w_i^2 </math>λ∥w∥2=λ∑i=1nwi2,也就是 L2正则化项

模型训练的过程就是寻找让损失函数 <math xmlns="http://www.w3.org/1998/Math/MathML"> L ( w ) L(w) </math>L(w)最小的参数 <math xmlns="http://www.w3.org/1998/Math/MathML"> w w </math>w。

也就等价于:\begin{align} & arg\ min(y-wX)^2 \\ & s.t. \sum w_{ij}^2 < s \end{align}

这两个公式表示,在满足约束条件 <math xmlns="http://www.w3.org/1998/Math/MathML"> ∑ w i j 2 < s \sum w_{ij}^2 < s </math>∑wij2<s的情况下,计算 <math xmlns="http://www.w3.org/1998/Math/MathML"> ( y − w X ) 2 (y-wX)^2 </math>(y−wX)2的最小值。

2. 创建样本数据

岭回归 适用于特征之间有很高关联性的数据集。

所以用scikit-learn中的加州住房数据集 ,这个数据集有8个房屋售价相关的属性,属性之间关联性高。

数据集的文件获取可以参考:sklearn基础--『数据加载』之真实数据集

从上面的文章中下载数据集(是一个zip压缩文件),

如下例所示,下载之后在 D:\share\data 中解压,就可以加载了。

python 复制代码
import os
from sklearn.datasets import fetch_california_housing

home_dir = "D:\share\data"
data = fetch_california_housing(data_home=os.path.join(home_dir, "cal_housing"))
X = data["data"]
y = data["target"]

大约有2万多条数据。

3. 模型训练

数据加载之后,首先划分训练集和测试集。

python 复制代码
from sklearn.model_selection import train_test_split

# 分割训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1)

然后用岭回归模型训练数据:

python 复制代码
from sklearn.linear_model import Ridge

# 初始化岭回归线性模型
reg = Ridge()
# 训练模型
reg.fit(X_train, y_train)

这里,用的Ridge()模型的默认参数,它的一些主要参数如下(训练模型时可根据情况调整参数):

  1. alpha :控制正则化强度的常量,也就是上面公式中的 <math xmlns="http://www.w3.org/1998/Math/MathML"> λ \lambda </math>λ,默认值1,设置为0时,就是最小二乘法
  2. fit_intercept :是否拟合此模型的截距,默认 True
  3. copy_X :是否复制X(也就是训练数据),默认 True,设置为False的话,有可能会改变训练数据
  4. tol:算法迭代时,收敛的精度上限
  5. solver :迭代时使用的求解器,包含** {auto, svd, cholesky, lsqr, sparse_cg, sag, saga, lbfgs}** 等算法,默认 auto(根据数据类型自动选择求解器)

最后,用测试数据来验证训练后模型的性能。

python 复制代码
y_pred = reg.predict(X_test)
mse = metrics.mean_squared_error(y_test, y_pred)
r2 = metrics.r2_score(y_test, y_pred)
m_error = metrics.median_absolute_error(y_test, y_pred)

print("均方误差:{}".format(mse))
print("复相关系数:{}".format(r2))
print("中位数绝对误差:{}".format(m_error))

# 运行结果
均方误差:0.0029948538129997903
复相关系数:0.9987534427417275
中位数绝对误差:0.049467455621301726

从结果来看,模型的性能还不错,均方误差中位数绝对误差 都比较小,而复相关系数高,说明在测试数据中,预测的值和实际的值比较接近。

4. 总结

总之,岭回归 在很多场景下都有应用,例如多元线性回归、时间序列预测、特征选择等。

它的主要优点是可以处理共线性数据,并且在加入噪声的情况下会有更稳定的性能。

然而,由于其对数据的缩放敏感岭回归 的一个主要局限性是它可能对数据的尺度非常敏感

此外,岭回归 正则化参数的选择通常需要一些经验或者实验来确定,这也增加了其应用的复杂性

PS.
共线性是指特征之间存在高度相关性,这可能导致线性回归模型的不稳定。

相关推荐
IT古董7 小时前
【漫话机器学习系列】249.Word2Vec自然语言训练模型
机器学习·自然语言处理·word2vec
白光白光8 小时前
大语言模型训练的两个阶段
人工智能·机器学习·语言模型
BioRunYiXue9 小时前
一文了解氨基酸的分类、代谢和应用
人工智能·深度学习·算法·机器学习·分类·数据挖掘·代谢组学
IT古董10 小时前
【漫话机器学习系列】255.独立同分布(Independent and Identically Distributed,简称 IID)
人工智能·机器学习
fytianlan10 小时前
机器学习 day6 -线性回归练习
人工智能·机器学习·线性回归
Blossom.11812 小时前
低代码开发:开启软件开发的新篇章
人工智能·深度学习·安全·低代码·机器学习·计算机视觉·数据挖掘
安特尼12 小时前
招行数字金融挑战赛数据赛道赛题一
人工智能·python·机器学习·金融·数据分析
极小狐13 小时前
如何构建容器镜像并将其推送到极狐GitLab容器镜像库?
开发语言·数据库·机器学习·gitlab·ruby
正在走向自律13 小时前
从0到1:Python机器学习实战全攻略(8/10)
开发语言·python·机器学习
北温凉15 小时前
【学习笔记】机器学习(Machine Learning) | 第六章(2)| 过拟合问题
笔记·机器学习