使用sklearn做单机特征工程
本文转载自 http://www.cnblogs.com/jasonfreak/p/5448385.html ,在某些部分会拓展补充一些内容,全文主要讲述有关特征工程中通常使用的方法以及在sklearn中的相关实现。
特征工程是什么
有这么一句话在业界广泛流传:数据和特征决定了机器学习的上限,而模型和算法只是逼近这个上限而已。那特征工程到底是什么呢?顾名思义,其本质是一项工程活动,目的是最大限度地从原始数据中提取特征以供算法和模型使用。通过总结和归纳,人们认为特征工程包括以下方面:

特征处理是特征工程的核心部分,sklearn提供了较为完整的特征处理方法,包括数据预处理,特征选择,降维等。首次接触到sklearn,通常会被其丰富且方便的算法模型库吸引,但是这里介绍的特征处理库也十分强大!
本文中使用sklearn中的IRIS(鸢尾花)数据集来对特征处理功能进行说明。IRIS数据集由Fisher在1936年整理,包含4个特征(Sepal.Length(花萼长度)、Sepal.Width(花萼宽度)、Petal.Length(花瓣长度)、Petal.Width(花瓣宽度)),特征值都为正浮点数,单位为厘米。目标值为鸢尾花的分类(Iris
Setosa(山鸢尾)、Iris Versicolour(杂色鸢尾),Iris
Virginica(维吉尼亚鸢尾))。导入IRIS数据集的代码如下:
1
2
3
4
5
6
7
8
9
10from sklearn.datasets import load_iris
#导入IRIS数据集
iris = load_iris()
#特征矩阵
iris.data
#目标向量
iris.target
数据预处理
通过特征提取,我们能得到未经处理的特征,这时的特征可能有以下问题:
- 不属于同一量纲:即特征的规格不一样,不能够放在一起比较。无量纲化可以解决这一问题。
- 信息冗余:对于某些定量特征,其包含的有效信息为区间划分,例如学习成绩,假若只关心“及格”或不“及格”,那么需要将定量的考分,转换成“1”和“0”表示及格和未及格。二值化可以解决这一问题。
- 定性特征不能直接使用:某些机器学习算法和模型只能接受定量特征的输入,那么需要将定性特征转换为定量特征。最简单的方式是为每一种定性值指定一个定量值,但是这种方式过于灵活,增加了调参的工作。通常使用哑编码的方式将定性特征转换为定量特征:假设有N种定性值,则将这一个特征扩展为N种特征,当原始特征值为第i种定性值时,第i个扩展特征赋值为1,其他扩展特征赋值为0。哑编码的方式相比直接指定的方式,不用增加调参的工作,对于线性模型来说,使用哑编码后的特征可达到非线性的效果。
- 存在缺失值:缺失值需要补充。
- 信息利用率低:不同的机器学习算法和模型对数据中信息的利用是不同的,之前提到在线性模型中,使用对定性特征哑编码可以达到非线性的效果。类似地,对定量变量多项式化,或者进行其他的转换,都能达到非线性的效果。
我们使用sklearn中的 preproccessing
库来进行数据预处理,可以覆盖以上问题的解决方案。
无量纲化
无量纲化使不同规格的数据转换到同一规格。常见的无量纲化方法有标准化和归一化。
标准化一般指的是0均值标准化(zero-mean normalization)也叫z-score标准化。z-score标准化的前提是特征值服从正态分布,标准化后,其转换成标准正态分布,即均值为0,标准差为1。
归一化一般指的是区间缩放法,利用了边界值信息,将特征的取值区间缩放到某个特点的范围,例如[0, 1]等。
标准化
0均值标准化需要计算特征的均值和标准差,公式表达为:\[x' = \frac{x - \mu}{\sigma}\] 公式中的 \(\mu\) 为原始数据的期望, \(\sigma\)为原始数据的标准差。
使用 preproccessing
库的 StandardScaler
类对数据进行标准化的代码如下:
1 | from sklearn.preprocessing import StandardScaler |
标准化的原理比较复杂,它表示的是原始值与均值之间差多少个标准差,是一个相对值,所以也有去除量纲的功效。同时,它还带来两个附加的好处:均值为0,标准差为1。关于这两个属性的具体好处参考 http://www.zhaokv.com/2016/01/normalization-and-standardization.html 如下
均值为0有什么好处呢?它可以使数据以0为中心左右分布(这不是废话嘛),而数据以0为中心左右分布会带来很多便利。比如在去中心化的数据上做SVD分解等价于在原始数据上做PCA;机器学习中很多函数如Sigmoid、Tanh、Softmax等都以0为中心左右分布(不一定对称)。
标准差为1有什么好处呢?这个更复杂一些。对于\(x_i, x_{i'}\)两点间距离,往往表示为 \[D(x_i,x_{i'})=\sum\limits_{j=1}^pw_j\cdot d_j(x_{ij},x_{i'j})(\sum\limits_{j=1}^pw_j=1)\] 其中 \(d_j(x_{ij},x_{i'j})\)是属性 \(j\) 两个点之间的距离, \(w_j\) 是该属性间距离在总距离中的权重 注意设 \(w_j=1,\forall j\)并不能实现每个属性对最后的结果贡献度相同。对于给定的数据集,所有点对间距离的平均值是个定值,即 \[\bar{D}=\frac{1}{N^2}\sum\limits_{i=1}^N\sum\limits_{i'=1}^ND(x_i,x_{i'})=\sum\limits_{j=1}^pw_j\cdot \bar{d}_j\] 是个常数,其中 \(\bar{d}_j=\frac{1}{N^2}\sum\limits_{i=1}^N\sum\limits_{i'=1}^Nd_j(x_{ij}, x_{x'j})\), 可见第 \(j\) 个变量对最终整体平均距离的影响是 \(w_j\cdot \bar{d}_j\),所以设 \(w_j\sim 1/\bar{d}_j\) 可以使所有属性对全体数据集平均距离的贡献相同。现在设 \(d_j\) 为欧氏距离的平方,它是最常用的距离衡量方法之一,则有\[\bar{d_j}=\frac{1}{N^2}\sum\limits_{i=1}^N\sum\limits_{i'=1}^N(x_{ij}-x_{i'j})^2=2\cdot var_j\]其中 \(var_j\) 是 \(Var(X_j)\) 样本估计,也就是说每个变量的重要程度正比于这个变量在这个数据集上的方差。如果我们让每一维变量的标准差都为1(即方差都为1),每维变量在计算距离的时候重要程度相同。
因此在分类、聚类算法中,需要使用距离来度量相似性的时候、或者使用PCA技术进行降维的时候,往往采用标准化方法。
归一化
区间缩放法的思路有多种,常见的一种为利用两个最值进行缩放,公式表达为:\[x' = \frac{x - Min}{Max - Min}\] 使用
preproccessing
库的 MinMaxScaler
类对数据进行区间缩放的代码如下:
1 | from sklearn.preprocessing import MinMaxScaler |
归一化的依据非常简单,不同变量往往量纲不同,归一化可以消除量纲对最终结果的影响,使不同变量具有可比性。比如两个人体重差10KG,身高差0.02M,在衡量两个人的差别时体重的差距会把身高的差距完全掩盖,归一化之后就不会有这样的问题。
除此之外,归一化后能够加快梯度下降求最优解的速度,如下图所示,蓝色的圈圈图代表的是两个特征的等高线。其中左图两个特征 \(X1\) 和 \(X2\) 的区间相差非常大,\(X1\) 区间是[0,2000],\(X2\)区间是[1,5],其所形成的等高线非常尖。当使用梯度下降法寻求最优解时,很有可能走“之字型”路线(垂直等高线走),从而导致需要迭代很多次才能收敛;而右图对两个原始特征进行了归一化,其对应的等高线显得很圆,在梯度下降进行求解时能较快的收敛。因此如果机器学习模型使用梯度下降法求最优解时,归一化往往非常有必要,否则很难收敛甚至不能收敛。

因此。在涉及到计算点与点之间的距离时,使用归一化或标准化都会对最后的结果有所提升,甚至会有质的区别。那在归一化与标准化之间应该如何选择呢?根据上面论述我们看到,如果把所有维度的变量一视同仁,在最后计算距离中发挥相同的作用应该选择标准化,如果想保留原始数据中由标准差所反映的潜在权重关系应该选择归一化。另外,标准化更适合现代嘈杂大数据场景。
对定量特征二值化
定量特征二值化的核心在于设定一个阈值,大于阈值的赋值为1,小于等于阈值的赋值为0。使用
preproccessing
库的 Binarizer
类对数据进行二值化的代码如下:
1 | from sklearn.preprocessing import Binarizer |
对定性特征哑编码
由于IRIS数据集的特征皆为定量特征,故使用其目标值进行哑编码(实际上是不需要的)。使用
preproccessing
库的 OneHotEncoder
类对数据进行哑编码的代码如下:
1 | from sklearn.preprocessing import OneHotEncoder |
缺失值计算
由于IRIS数据集没有缺失值,故对数据集新增一个样本,4个特征均赋值为NaN,表示数据缺失。使用
preproccessing
库的 Imputer
类对数据进行缺失值计算的代码如下:
1 | from numpy import vstack, array, nan |
数据变换
数据变换就是通过组合或变换方式将原来的特征转换为新的特征,常见的数据变换有基于多项式的、基于指数函数的、基于对数函数的。
例如,输入的一个二维特征为 \([a, b]\),则度为2的多项式特征为 $ [1, a, b, a^2, ab, b^2]$
使用 preproccessing
库的 PolynomialFeatures
类对数据进行多项式转换的代码如下:
1 | from sklearn.preprocessing import PolynomialFeatures |
基于单变元函数的数据变换可以使用一个统一的方式完成,使用
preproccessing
库的 FunctionTransformer
对数据进行对数函数转换的代码如下:
1 | from numpy import log1p |
小结
类 | 功能 | 说明 |
---|---|---|
StandardScaler | 无量纲化 | 标准化,基于特征矩阵的列,将特征值转换至服从标准正态分布 |
MinMaxScaler | 无量纲化 | 区间缩放,基于最大最小值,将特征值转换到[0, 1]区间上 |
Normalizer | 归一化 | 基于特征矩阵的行,将样本向量转换为“单位向量” |
Binarizer | 二值化 | 基于给定阈值,将定量特征按阈值划分 |
OneHotEncoder | 哑编码 | 将定性数据编码为定量数据 |
Imputer | 缺失值计算 | 计算缺失值,缺失值可填充为均值等 |
PolynomialFeatures | 多项式数据转换 | 多项式数据转换 |
FunctionTransformer | 自定义单元数据转换 | 使用单变元的函数来转换数据 |
特征选择
当数据预处理完成后,我们需要选择有意义的特征输入机器学习的算法和模型进行训练。通常来说,从两个方面考虑来选择特征:
- 特征是否发散:如果一个特征不发散,例如方差接近于0,也就是说样本在这个特征上基本上没有差异,这个特征对于样本的区分并没有什么用。
- 特征与目标的相关性:这点比较显见,与目标相关性高的特征,应当优选选择。除方差法外,本文介绍的其他方法均从相关性考虑。
根据特征选择的形式又可以将特征选择方法分为3种:
- Filter:过滤法,按照发散性或者相关性对各个特征进行评分,设定阈值或者待选择阈值的个数,选择特征。
- Wrapper:包装法,根据目标函数(通常是预测效果评分),每次选择若干特征,或者排除若干特征。
- Embedded:嵌入法,先使用某些机器学习的算法和模型进行训练,得到各个特征的权值系数,根据系数从大到小选择特征。类似于Filter方法,但是是通过训练来确定特征的优劣。
我们使用sklearn中的feature_selection库来进行特征选择。
Filter
方差选择法
使用方差选择法,先要计算各个特征的方差,然后根据阈值,选择方差大于阈值的特征。使用
feature_selection
库的 VarianceThreshold
类来选择特征的代码如下: 1
2
3
4
5from sklearn.feature_selection import VarianceThreshold
#方差选择法,返回值为特征选择后的数据
#参数threshold为方差的阈值
VarianceThreshold(threshold=3).fit_transform(iris.data)
相关系数法
使用相关系数法,先要计算各个特征对目标值的相关系数以及相关系数的P值。用
feature_selection
库的 SelectKBest
类结合相关系数来选择特征的代码如下:
1 | from sklearn.feature_selection import SelectKBest |
卡方检验
经典的卡方检验是检验定性自变量对定性因变量的相关性。具体的意义可参考这篇文章卡方检验原理及应用。用
feature_selection
库的 SelectKBest
类结合卡方检验来选择特征的代码如下
1 | from sklearn.feature_selection import SelectKBest |
互信息法
经典的互信息也是变量间相互依赖性的量度,互信息计算公式如下: \[I(X,Y) = \sum_{x \in X}\sum_{y \in
Y}p(x,y)log\frac{p(x,y)}{p(x)p(y)}\]
为了处理定量数据,最大信息系数法被提出,使用
feature_selection
库的 SelectKBest
类结合最大信息系数法来选择特征的代码如下:
1 | from sklearn.feature_selection import SelectKBest |
Wrapper
递归特征消除法
递归消除特征法使用一个基模型来进行多轮训练,每轮训练后,消除若干权值系数的特征,再基于新的特征集进行下一轮训练。使用feature_selection库的RFE类来选择特征的代码如下:
1
2
3
4
5
6
7from sklearn.feature_selection import RFE
from sklearn.linear_model import LogisticRegression
#递归特征消除法,返回特征选择后的数据
#参数estimator为基模型
#参数n_features_to_select为选择的特征个数
RFE(estimator=LogisticRegression(), n_features_to_select=2).fit_transform(iris.data, iris.target)
Embedded
基于惩罚项的特征选择法
使用带惩罚项的基模型,除了筛选出特征外,同时也进行了降维。使用feature_selection
库的SelectFromModel
类结合带L1惩罚项的逻辑回归模型,来选择特征的代码如下
1 | from sklearn.feature_selection import SelectFromModel |
L1惩罚项降维的原理在于保留多个对目标值具有同等相关性的特征中的一个,所以没选到的特征不代表不重要。故,可结合L2惩罚项来优化。具体操作为:若一个特征在L1中的权值为1,选择在L2中权值差别不大且在L1中权值为0的特征构成同类集合,将这一集合中的特征平分L1中的权值,故需要构建一个新的逻辑回归模型:
1 | from sklearn.linear_model import LogisticRegression |
使用feature_selection
库的SelectFromModel
类结合带L1以及L2惩罚项的逻辑回归模型,来选择特征的代码如下:
1 | from sklearn.feature_selection import SelectFromModel |
基于树模型的特征选择法
树模型中GBDT也可用来作为基模型进行特征选择,使用
feature_selection
库的 SelectFromModel
类结合
GBDT 模型,来选择特征的代码如下:
1 | from sklearn.feature_selection import SelectFromModel |
小结
类 | 所属方式 | 说明 |
---|---|---|
VarianceThreshold | Filter | 方差选择法 |
SelectKBest | Filter | 可选关联系数、卡方校验、最大信息系数作为得分计算的方法 |
RFE | Wrapper | 递归地训练基模型,将权值系数较小的特征从特征集合中消除 |
SelectFromModel | Embedded | 训练基模型,选择权值系数较高的特征 |
降维
当特征选择完成后,可以直接训练模型了,但是可能由于特征矩阵过大,导致计算量大,训练时间长的问题,因此降低特征矩阵维度也是必不可少的。常见的降维方法除了以上提到的基于L1惩罚项的模型以外,另外还有主成分分析法(PCA)和线性判别分析(LDA),线性判别分析本身也是一个分类模型。PCA和LDA有很多的相似点,其本质是要将原始的样本映射到维度更低的样本空间中,但是PCA和LDA的映射目标不一样:PCA是为了让映射后的样本具有最大的发散性;而LDA是为了让映射后的样本有最好的分类性能。所以说PCA是一种无监督的降维方法,而LDA是一种有监督的降维方法。
主成分分析法(PCA)
使用 decomposition
库的PCA类选择特征的代码如下:
1
2
3
4
5from sklearn.decomposition import PCA
#主成分分析法,返回降维后的数据
#参数n_components为主成分数目
PCA(n_components=2).fit_transform(iris.data)
线性判别分析法(LDA)
使用lda库的LDA类选择特征的代码如下: 1
2
3
4
5from sklearn.lda import LDA
#线性判别分析法,返回降维后的数据
#参数n_components为降维后的维数
LDA(n_components=2).fit_transform(iris.data, iris.target)
小结
库 | 类 | 说明 |
---|---|---|
decomposition | PCA | 主成分分析法 |
lda | LDA | 线性判别分析法 |
总结
再让我们回归一下本文开始的特征工程的思维导图,我们可以使用sklearn完成几乎所有特征处理的工作,而且不管是数据预处理,还是特征选择,抑或降维,它们都是通过某个类的方法fit_transform
完成的,fit_transform
要不只带一个参数:特征矩阵,要不带两个参数:特征矩阵加目标向量。这些难道都是巧合吗?还是故意设计成这样?方法fit_transform
中有fit这一单词,它和训练模型的fit方法有关联吗?接下来,在《使用sklearn优雅地进行数据挖掘》中将会阐述其中的奥妙!