本文轉載

使用sklearn做單機特征工程

目錄

1 特征工程是什么？
2 數據預處理
　　2.1 無量綱化
　　　　2.1.1 標準化
　　　　2.1.2 區間縮放法
　　　　2.1.3 標準化與歸一化的區別
　　2.2 對定量特征二值化
　　2.3 對定性特征啞編碼
　　2.4 缺失值計算
　　2.5 數據變換
　　2.6 回顧
3 特征選擇
　　3.1 Filter
　　　　3.1.1 方差選擇法
　　　　3.1.2 相關系數法
　　　　3.1.3 卡方檢驗
　　　　3.1.4 互信息法
　　3.2 Wrapper
　　　　3.2.1 遞歸特征消除法
　　3.3 Embedded
　　　　3.3.1 基于懲罰項的特征選擇法
　　　　3.3.2 基于樹模型的特征選擇法
　　3.4 回顧
4 降維
　　4.1 主成分分析法（PCA）
　　4.2 線性判別分析法（LDA）
　　4.3 回顧
5 總結
6 參考資料

---

1 特征工程是什么？

　　有這么一句話在業界廣泛流傳：數據和特征決定了機器學習的上限，而模型和算法只是逼近這個上限而已。那特征工程到底是什么呢？顧名思義，其本質是一項工程活動，目的是最大限度地從原始數據中提取特征以供算法和模型使用。通過總結和歸納，人們認為特征工程包括以下方面：

　　特征處理是特征工程的核心部分，sklearn提供了較為完整的特征處理方法，包括數據預處理，特征選擇，降維等。首次接觸到sklearn，通常會被其豐富且方便的算法模型庫吸引，但是這里介紹的特征處理庫也十分強大！

　　本文中使用sklearn中的IRIS（鳶尾花）數據集來對特征處理功能進行說明。IRIS數據集由Fisher在1936年整理，包含4個特征（Sepal.Length（花萼長度）、Sepal.Width（花萼寬度）、Petal.Length（花瓣長度）、Petal.Width（花瓣寬度）），特征值都為正浮點數，單位為厘米。目標值為鳶尾花的分類（Iris Setosa（山鳶尾）、Iris Versicolour（雜色鳶尾），Iris Virginica（維吉尼亞鳶尾））。導入IRIS數據集的代碼如下：

1 from sklearn.datasets import load_iris2 3 #導入IRIS數據集4 iris = load_iris()5 6 #特征矩陣7 iris.data8 9 #目標向量 10 iris.target

---

2 數據預處理

　　通過特征提取，我們能得到未經處理的特征，這時的特征可能有以下問題：

• 不屬于同一量綱：即特征的規格不一樣，不能夠放在一起比較。無量綱化可以解決這一問題。
• 信息冗余：對于某些定量特征，其包含的有效信息為區間劃分，例如學習成績，假若只關心“及格”或不“及格”，那么需要將定量的考分，轉換成“1”和“0”表示及格和未及格。二值化可以解決這一問題。
• 定性特征不能直接使用：某些機器學習算法和模型只能接受定量特征的輸入，那么需要將定性特征轉換為定量特征。最簡單的方式是為每一種定性值指定一個定量值，但是這種方式過于靈活，增加了調參的工作。通常使用啞編碼的方式將定性特征轉換為定量特征：假設有N種定性值，則將這一個特征擴展為N種特征，當原始特征值為第i種定性值時，第i個擴展特征賦值為1，其他擴展特征賦值為0。啞編碼的方式相比直接指定的方式，不用增加調參的工作，對于線性模型來說，使用啞編碼后的特征可達到非線性的效果。
• 存在缺失值：缺失值需要補充。
• 信息利用率低：不同的機器學習算法和模型對數據中信息的利用是不同的，之前提到在線性模型中，使用對定性特征啞編碼可以達到非線性的效果。類似地，對定量變量多項式化，或者進行其他的轉換，都能達到非線性的效果。

　　我們使用sklearn中的preproccessing庫來進行數據預處理，可以覆蓋以上問題的解決方案。

2.1 無量綱化

　　無量綱化使不同規格的數據轉換到同一規格。常見的無量綱化方法有標準化和區間縮放法。標準化的前提是特征值服從正態分布，標準化后，其轉換成標準正態分布。區間縮放法利用了邊界值信息，將特征的取值區間縮放到某個特點的范圍，例如[0, 1]等。

2.1.1 標準化

　　標準化需要計算特征的均值和標準差，公式表達為：

　　使用preproccessing庫的StandardScaler類對數據進行標準化的代碼如下：

1 from sklearn.preprocessing import StandardScaler 2 3 #標準化，返回值為標準化后的數據 4 StandardScaler().fit_transform(iris.data)

2.1.2 區間縮放法

　　區間縮放法的思路有多種，常見的一種為利用兩個最值進行縮放，公式表達為：

　　使用preproccessing庫的MinMaxScaler類對數據進行區間縮放的代碼如下：

1 from sklearn.preprocessing import MinMaxScaler 2 3 #區間縮放，返回值為縮放到[0, 1]區間的數據 4 MinMaxScaler().fit_transform(iris.data)

2.1.3 標準化與歸一化的區別

　　簡單來說，標準化是依照特征矩陣的列處理數據，其通過求z-score的方法，將樣本的特征值轉換到同一量綱下。歸一化是依照特征矩陣的行處理數據，其目的在于樣本向量在點乘運算或其他核函數計算相似性時，擁有統一的標準，也就是說都轉化為“單位向量”。規則為l2的歸一化公式如下：

　　使用preproccessing庫的Normalizer類對數據進行歸一化的代碼如下：

1 from sklearn.preprocessing import Normalizer 2 3 #歸一化，返回值為歸一化后的數據 4 Normalizer().fit_transform(iris.data)

2.2 對定量特征二值化

　　定量特征二值化的核心在于設定一個閾值，大于閾值的賦值為1，小于等于閾值的賦值為0，公式表達如下：

　　使用preproccessing庫的Binarizer類對數據進行二值化的代碼如下：

1 from sklearn.preprocessing import Binarizer 2 3 #二值化，閾值設置為3，返回值為二值化后的數據 4 Binarizer(threshold=3).fit_transform(iris.data)

2.3 對定性特征啞編碼

　　由于IRIS數據集的特征皆為定量特征，故使用其目標值進行啞編碼（實際上是不需要的）。使用preproccessing庫的OneHotEncoder類對數據進行啞編碼的代碼如下：

1 from sklearn.preprocessing import OneHotEncoder 2 3 #啞編碼，對IRIS數據集的目標值，返回值為啞編碼后的數據 4 OneHotEncoder().fit_transform(iris.target.reshape((-1,1)))

2.4 缺失值計算

　　由于IRIS數據集沒有缺失值，故對數據集新增一個樣本，4個特征均賦值為NaN，表示數據缺失。使用preproccessing庫的Imputer類對數據進行缺失值計算的代碼如下：

1 from numpy import vstack, array, nan 2 from sklearn.preprocessing import Imputer 3 4 #缺失值計算，返回值為計算缺失值后的數據 5 #參數missing_value為缺失值的表示形式，默認為NaN 6 #參數strategy為缺失值填充方式，默認為mean（均值） 7 Imputer().fit_transform(vstack((array([nan, nan, nan, nan]), iris.data)))

2.5 數據變換

　　常見的數據變換有基于多項式的、基于指數函數的、基于對數函數的。4個特征，度為2的多項式轉換公式如下：

　　使用preproccessing庫的PolynomialFeatures類對數據進行多項式轉換的代碼如下：

1 from sklearn.preprocessing import PolynomialFeatures 2 3 #多項式轉換 4 #參數degree為度，默認值為2 5 PolynomialFeatures().fit_transform(iris.data)

　　基于單變元函數的數據變換可以使用一個統一的方式完成，使用preproccessing庫的FunctionTransformer對數據進行對數函數轉換的代碼如下：

1 from numpy import log1p 2 from sklearn.preprocessing import FunctionTransformer 3 4 #自定義轉換函數為對數函數的數據變換 5 #第一個參數是單變元函數 6 FunctionTransformer(log1p).fit_transform(iris.data)

2.6 回顧

類	功能	說明
StandardScaler	無量綱化	標準化，基于特征矩陣的列，將特征值轉換至服從標準正態分布
MinMaxScaler	無量綱化	區間縮放，基于最大最小值，將特征值轉換到[0, 1]區間上
Normalizer	歸一化	基于特征矩陣的行，將樣本向量轉換為“單位向量”
Binarizer	二值化	基于給定閾值，將定量特征按閾值劃分
OneHotEncoder	啞編碼	將定性數據編碼為定量數據
Imputer	缺失值計算	計算缺失值，缺失值可填充為均值等
PolynomialFeatures	多項式數據轉換	多項式數據轉換
FunctionTransformer	自定義單元數據轉換	使用單變元的函數來轉換數據

?

---

?

3 特征選擇

　　當數據預處理完成后，我們需要選擇有意義的特征輸入機器學習的算法和模型進行訓練。通常來說，從兩個方面考慮來選擇特征：

• 特征是否發散：如果一個特征不發散，例如方差接近于0，也就是說樣本在這個特征上基本上沒有差異，這個特征對于樣本的區分并沒有什么用。
• 特征與目標的相關性：這點比較顯見，與目標相關性高的特征，應當優選選擇。除方差法外，本文介紹的其他方法均從相關性考慮。

　　根據特征選擇的形式又可以將特征選擇方法分為3種：

• Filter：過濾法，按照發散性或者相關性對各個特征進行評分，設定閾值或者待選擇閾值的個數，選擇特征。
• Wrapper：包裝法，根據目標函數（通常是預測效果評分），每次選擇若干特征，或者排除若干特征。
• Embedded：嵌入法，先使用某些機器學習的算法和模型進行訓練，得到各個特征的權值系數，根據系數從大到小選擇特征。類似于Filter方法，但是是通過訓練來確定特征的優劣。

　　我們使用sklearn中的feature_selection庫來進行特征選擇。

3.1 Filter

3.1.1 方差選擇法

　　使用方差選擇法，先要計算各個特征的方差，然后根據閾值，選擇方差大于閾值的特征。使用feature_selection庫的VarianceThreshold類來選擇特征的代碼如下：

1 from sklearn.feature_selection import VarianceThreshold 2 3 #方差選擇法，返回值為特征選擇后的數據 4 #參數threshold為方差的閾值 5 VarianceThreshold(threshold=3).fit_transform(iris.data)

3.1.2 相關系數法

　　使用相關系數法，先要計算各個特征對目標值的相關系數以及相關系數的P值。用feature_selection庫的SelectKBest類結合相關系數來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectKBest 2 from scipy.stats import pearsonr 3 4 #選擇K個最好的特征，返回選擇特征后的數據 5 #第一個參數為計算評估特征是否好的函數，該函數輸入特征矩陣和目標向量，輸出二元組（評分，P值）的數組，數組第i項為第i個特征的評分和P值。在此定義為計算相關系數 6 #參數k為選擇的特征個數 7 SelectKBest(lambda X, Y: array(map(lambda x:pearsonr(x, Y), X.T)).T, k=2).fit_transform(iris.data, iris.target)

3.1.3 卡方檢驗

　　經典的卡方檢驗是檢驗定性自變量對定性因變量的相關性。假設自變量有N種取值，因變量有M種取值，考慮自變量等于i且因變量等于j的樣本頻數的觀察值與期望的差距，構建統計量：

　　這個統計量的含義簡而言之就是自變量對因變量的相關性。用feature_selection庫的SelectKBest類結合卡方檢驗來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectKBest 2 from sklearn.feature_selection import chi2 3 4 #選擇K個最好的特征，返回選擇特征后的數據 5 SelectKBest(chi2, k=2).fit_transform(iris.data, iris.target)

3.1.4 互信息法

　　經典的互信息也是評價定性自變量對定性因變量的相關性的，互信息計算公式如下：

　　為了處理定量數據，最大信息系數法被提出，使用feature_selection庫的SelectKBest類結合最大信息系數法來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectKBest2 from minepy import MINE3 4 #由于MINE的設計不是函數式的，定義mic方法將其為函數式的，返回一個二元組，二元組的第2項設置成固定的P值0.55 def mic(x, y):6 m = MINE()7 m.compute_score(x, y)8 return (m.mic(), 0.5)9 10 #選擇K個最好的特征，返回特征選擇后的數據 11 SelectKBest(lambda X, Y: array(map(lambda x:mic(x, Y), X.T)).T, k=2).fit_transform(iris.data, iris.target)

3.2 Wrapper

3.2.1 遞歸特征消除法

　　遞歸消除特征法使用一個基模型來進行多輪訓練，每輪訓練后，消除若干權值系數的特征，再基于新的特征集進行下一輪訓練。使用feature_selection庫的RFE類來選擇特征的代碼如下：

1 from sklearn.feature_selection import RFE 2 from sklearn.linear_model import LogisticRegression 3 4 #遞歸特征消除法，返回特征選擇后的數據 5 #參數estimator為基模型 6 #參數n_features_to_select為選擇的特征個數 7 RFE(estimator=LogisticRegression(), n_features_to_select=2).fit_transform(iris.data, iris.target)

3.3 Embedded

3.3.1 基于懲罰項的特征選擇法

　　使用帶懲罰項的基模型，除了篩選出特征外，同時也進行了降維。使用feature_selection庫的SelectFromModel類結合帶L1懲罰項的邏輯回歸模型，來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectFromModel 2 from sklearn.linear_model import LogisticRegression 3 4 #帶L1懲罰項的邏輯回歸作為基模型的特征選擇 5 SelectFromModel(LogisticRegression(penalty="l1", C=0.1)).fit_transform(iris.data, iris.target)

　　L1懲罰項降維的原理在于保留多個對目標值具有同等相關性的特征中的一個，所以沒選到的特征不代表不重要。故，可結合L2懲罰項來優化。具體操作為：若一個特征在L1中的權值為1，選擇在L2中權值差別不大且在L1中權值為0的特征構成同類集合，將這一集合中的特征平分L1中的權值，故需要構建一個新的邏輯回歸模型：

?View Code

　　使用feature_selection庫的SelectFromModel類結合帶L1以及L2懲罰項的邏輯回歸模型，來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectFromModel 2 3 #帶L1和L2懲罰項的邏輯回歸作為基模型的特征選擇 4 #參數threshold為權值系數之差的閾值 5 SelectFromModel(LR(threshold=0.5, C=0.1)).fit_transform(iris.data, iris.target)

3.3.2 基于樹模型的特征選擇法

　　樹模型中GBDT也可用來作為基模型進行特征選擇，使用feature_selection庫的SelectFromModel類結合GBDT模型，來選擇特征的代碼如下：

1 from sklearn.feature_selection import SelectFromModel 2 from sklearn.ensemble import GradientBoostingClassifier 3 4 #GBDT作為基模型的特征選擇 5 SelectFromModel(GradientBoostingClassifier()).fit_transform(iris.data, iris.target)

3.4 回顧

類	所屬方式	說明
VarianceThreshold	Filter	方差選擇法
SelectKBest	Filter	可選關聯系數、卡方校驗、最大信息系數作為得分計算的方法
RFE	Wrapper	遞歸地訓練基模型，將權值系數較小的特征從特征集合中消除
SelectFromModel	Embedded	訓練基模型，選擇權值系數較高的特征

?

---

4 降維

　　當特征選擇完成后，可以直接訓練模型了，但是可能由于特征矩陣過大，導致計算量大，訓練時間長的問題，因此降低特征矩陣維度也是必不可少的。常見的降維方法除了以上提到的基于L1懲罰項的模型以外，另外還有主成分分析法（PCA）和線性判別分析（LDA），線性判別分析本身也是一個分類模型。PCA和LDA有很多的相似點，其本質是要將原始的樣本映射到維度更低的樣本空間中，但是PCA和LDA的映射目標不一樣：PCA是為了讓映射后的樣本具有最大的發散性；而LDA是為了讓映射后的樣本有最好的分類性能。所以說PCA是一種無監督的降維方法，而LDA是一種有監督的降維方法。

4.1 主成分分析法（PCA）

　　使用decomposition庫的PCA類選擇特征的代碼如下：

1 from sklearn.decomposition import PCA 2 3 #主成分分析法，返回降維后的數據 4 #參數n_components為主成分數目 5 PCA(n_components=2).fit_transform(iris.data)

4.2 線性判別分析法（LDA）

　　使用lda庫的LDA類選擇特征的代碼如下：

1 from sklearn.lda import LDA 2 3 #線性判別分析法，返回降維后的數據 4 #參數n_components為降維后的維數 5 LDA(n_components=2).fit_transform(iris.data, iris.target)

4.3 回顧

庫	類	說明
decomposition	PCA	主成分分析法
lda	LDA	線性判別分析法

?

---

5 總結

　　再讓我們回歸一下本文開始的特征工程的思維導圖，我們可以使用sklearn完成幾乎所有特征處理的工作，而且不管是數據預處理，還是特征選擇，抑或降維，它們都是通過某個類的方法fit_transform完成的，fit_transform要不只帶一個參數：特征矩陣，要不帶兩個參數：特征矩陣加目標向量。這些難道都是巧合嗎？還是故意設計成這樣？方法fit_transform中有fit這一單詞，它和訓練模型的fit方法有關聯嗎？接下來，我將在《使用sklearn優雅地進行數據挖掘》中闡述其中的奧妙！

---

6 參考資料

FAQ: What is dummy coding?

IRIS（鳶尾花）數據集

卡方檢驗

干貨：結合Scikit-learn介紹幾種常用的特征選擇方法

機器學習中，有哪些特征選擇的工程方法？

機器學習中的數學(4)-線性判別分析（LDA）, 主成分分析(PCA)

總結

以上是生活随笔為你收集整理的sklearn特征工程的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。