19、Python数据特征选择，模型选择法

 1  模型选择法

把建好的模型对象传入选择器，然后根据这个已经建好的模型，自动帮我们选择最好的特征值。

2 案例代码

import pandas

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


# 导入线性回归方程和模型选择方法


from sklearn.linear_model import LinearRegression

from sklearn.feature_selection import SelectFromModel


# 自变量特征

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

# 建立线性回归模型

LrModel= LinearRegression()

# 建立选择回归模型

selectFromModel= SelectFromModel(LrModel)

# fit方法训练选择，自动选择最优的特征数

selectFromModel.fit_transform(
        feature,
        data['销售额']
        )


feature.columns[selectFromModel.get_support()]