sklearn特征选择

最新推荐文章于 2024-08-02 16:27:08 发布

whytlok

最新推荐文章于 2024-08-02 16:27:08 发布

阅读量558

点赞数

分类专栏：人工智能文章标签： sklearn 机器学习 python

本文链接：https://blog.csdn.net/weixin_45320238/article/details/127478457

版权

人工智能专栏收录该内容

12 篇文章 0 订阅

订阅专栏

选取特征x中与标签y强相关的特征排名

def feature_SelectKBest(x,y):
    x = (x - x.min()) / (x.max() - x.min())# standardize data
    
    x.dropna(axis=1, how='all', thresh=None, subset=None, inplace=True)#clear the data being NaN
    bestfeatures = SelectKBest(score_func=f_regression, k=5)
    fit = bestfeatures.fit(x,y) 
    dfscores = pd.DataFrame(fit.scores_)
    dfcolumns = pd.DataFrame(x.columns)
    #concat two dataframes for better visualization
    featureScores = pd.concat([dfcolumns,dfscores],axis=1)
    featureScores.columns = ['Specs','Score']  #naming the dataframe columns
    myx=featureScores.nlargest(19,'Score').iloc[:,0].nsmallest(19)#统计前19个中最大的数的raw_x_train前19序号
    # 2,7,8,9,11,16,17,18,20,22,23,25,26,27,28,33,38,54,55#19
    for col in myx:
        print(col,end=",")