from sklearn.feature_selection import VarianceThreshold,SelectKBest,chi2
from sklearn.datasets import load_iris
import pandas as pd
X,y = load_iris(return_X_y=True)
X_df = pd.DataFrame(X,columns=list("ABCD"))
(chi2,pval) = chi2(X_df,y)
dict_feature = {}
for i,j in zip(X_df.columns.values,chi2):
dict_feature[i]=j
#对字典按照values排序
ls = sorted(dict_feature.items(),key=lambda item:item[1],reverse=True)
#特征选取数量
k =2
ls_new_feature=[]
for i in range(k):
ls_new_feature.append(ls[i][0])
X_new = X_df[ls_new_feature]
from sklearn.feature_selection import VarianceThreshold,SelectKBest,chi2
from sklearn.datasets import load_iris
import pandas as pd
from sklearn.feature_selection import mutual_info_classif
#用于度量特征和离散目标的互信息
X,y = load_iris(return_X_y=True)
X_df = pd.DataFrame(X,columns=list("ABCD"))
feature_cat = ["A","D"]
discrete_features = []
fe
python 特征筛选
最新推荐文章于 2024-01-20 14:26:59 发布
本文详细介绍了Python中如何进行特征筛选,包括基于统计学的方法如方差阈值、互信息、皮尔逊相关系数等,以及使用模型选择方法如递归特征消除(RFE)。通过实例展示了如何在实际数据集上应用这些方法,帮助提高模型的预测性能和理解力。
摘要由CSDN通过智能技术生成