【问题标题】:sklearn random forest to find score of selected featuressklearn随机森林查找所选特征的分数
【发布时间】:2020-08-19 08:18:16
【问题描述】:

我正在尝试了解每个所选特征获得的相关分数。

到目前为止我已经尝试过:

 classifier = 
 SelectFromModel(RandomForestClassifier(n_estimators = 100))
 m = classifier.fit(train.drop(columns='indicator'), train.rg_risk)
 X_train = train.drop(columns='indicator')
selected_feat=X_train.columns[(classifier.get_support())]
len(selected_feat)

【问题讨论】:

    标签: python machine-learning scikit-learn feature-selection


    【解决方案1】:

    SelectFromModel 是一种嵌入式方法:它使用具有内置特征选择方法的算法。

    在您的情况下,您使用 RandomForest 根据特征重要性选择特征。它使用每个决策树中的节点杂质来计算特征重要性。

    通过threshold=None,最终的特征重要性阈值默认计算为所有决策树特征重要性的平均值。其他可能性是中位数(与均值相同,但使用中位数)或用于调整中位数/均值的比例因子("1.25*mean""1.25*median")。

    source sklearn

    【讨论】:

      猜你喜欢
      • 2014-04-17
      • 2019-03-14
      • 2015-09-28
      • 2017-07-19
      • 2014-09-03
      • 2020-05-16
      • 2018-06-17
      • 2016-07-23
      • 2021-05-09
      相关资源
      最近更新 更多