1. 前向搜索
每次增量地从剩余未选中的特征选出一个加入特征集中,待达到阈值或者 时,从所有的
中选出错误率最小的。过程如下:
- 初始化特征集
为空。
- 扫描
从
到
- 如果第
个特征不在
中,那么特征
和
放在一起作为
(即
)。
- 在只使用
中特征的情况下,利用交叉验证来得到
的错误率。
- 从上步中得到的
个
中选出错误率最小的
,更新
为
。
- 如果
中的特征数达到了
或者预定的阈值(如果有的话),
- 那么输出整个搜索过程中最好的 ;若没达到,则转到 2,继续扫描。
2. 后向搜索
既然有增量加,那么也会有增量减,后者称为后向搜索。先将 设置为
,然后每次删除一个特征,并评价,直到达到阈值或者为空,然后选择最佳的
。
这两种算法都可以工作,但是计算复杂度比较大。时间复杂度为
3. 递归特征消除法
递归消除特征法使用一个基模型来进行多轮训练,每轮训练后,消除若干权值系数的特征,再基于新的特征集进行下一轮训练。使用feature_selection库的RFE类来选择特征的代码如下:
from 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)
举例:使用一个基模型来进行多轮训练,经过多轮训练后,保留指定的特征数。
#首先导入数据到data变量中 import pandas data=pandas.read_csv('路径.csv') #接着,我们使用RFE类,在estimator中, #把我们的基模型设置为线性回归模型LinearRegression, #然后在把我们要选择的特征数设置为2, #接着就可以使用这个rfe对象,把自变量和因变量传入fit_transform方法, #即可得到我们需要的特征值 from sklearn.feature_selection import RFE from sklearn.linear_model import LinearRegression feature =data[['月份','季度','广告推广费','注册并投资人数']] rfe =RFE( estimator=LinearRegression(), n_features_to_select=2 ) sFeature = rfe.fit_transform( feature, data['销售金额'] ) #同理,我们要想知道这两个自变量的名字, #使用get_support方法,即可得到对应的列名 rfe.get_support()