python保存模型 特征_19、Python数据特征选择，模型选择法

1  模型选择法

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

2 案例代码

[code]

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()]

阅读更多