18 python数据特征选择，递归特征消除法

1 递归特征消除法（RFE）

使用一个基模型来进行多轮训练，经过多轮训练后，保留置顶的特征数

RFE是recursive feature elimination回归特征消除，让回归特征消除过程中只保留no_features个最重要的特征，可以避免过度拟合，但RFE会舍弃一些变量，原没有下面几个方法给变量赋权重来的好。

2 代码案例

import pandas

data = pandas.read_csv('D:\\database\\pystudy\\6.2\\data2.csv')


# 导入RFE方法和线性回归基模型

from sklearn.feature_selection import RFE

from sklearn.linear_model import LinearRegression

# 自变量特征

feature=data[['月份','季度','广告费用','客流量']]

rfe=RFE(
        estimator=LinearRegression(), # 选择lin线性回归为基模型
        n_features_to_select=2   # 选区特征数 
        )

# fit 方法训练选择特征属性

sFeature=rfe.fit_transform(feature,data['销售额'])

feature.columns[rfe.get_support()] #查看满足条件的属性