SelectFromModel：如何调整阈值来控制特征选择的数量？

在使用SelectFromModel结合如梯度提升决策树（GBDT）等模型进行特征选择时，可以通过以下几种方式调整阈值来控制选择的特征数量：

一、设置阈值参数（以基于特征重要性为例）

1. 对于一些模型，如基于树的模型，通常会计算特征重要性。SelectFromModel可以根据这个重要性来选择特征。可以设置一个绝对的特征重要性阈值，只有重要性高于这个阈值的特征会被选择。
   • 示例代码：

   from sklearn.feature_selection import SelectFromModel
   from sklearn.ensemble import GradientBoostingClassifier
   
   gbdt = GradientBoostingClassifier()
   selector = SelectFromModel(gbdt, threshold=0.1)  # 设置特征重要性阈值为 0.1
   selected_train = selector.fit_transform(train, target_train)
   

   • 在这个例子中，只有特征重要性大于 0.1 的特征会被选择。可以通过调整这个阈值来控制选择的特征数量。阈值设置得越高，选择的特征数量可能越少。

二、使用百分位数作为阈值

1. 可以根据特征重要性的百分位数来设置阈值。例如，选择重要性排在前 50%的特征。
   • 示例代码：

   import numpy as np
   from sklearn.feature_selection import SelectFromModel
   from sklearn.ensemble import GradientBoostingClassifier
   
   gbdt = GradientBoostingClassifier()
   selector = SelectFromModel(gbdt)
   selector.fit(train, target_train)
   feature_importances = selector.estimator_.feature_importances_
   threshold = np.percentile(feature_importances, 50)  # 选择重要性排在前 50%的特征
   selector.threshold_ = threshold
   selected_train = selector.transform(train)
   

   • 这里首先使用SelectFromModel进行特征选择，然后计算特征重要性的百分位数作为阈值，并将其设置到selector对象中，最后进行特征选择。

三、通过交叉验证调整阈值

1. 使用交叉验证来评估不同阈值下的模型性能，选择性能最佳的阈值。
   • 示例代码：

   from sklearn.model_selection import cross_val_score
   from sklearn.feature_selection import SelectFromModel
   from sklearn.ensemble import GradientBoostingClassifier
   
   gbdt = GradientBoostingClassifier()
   selector = SelectFromModel(gbdt)
   
   for threshold in [0.05, 0.1, 0.15]:
       selector.threshold_ = threshold
       scores = cross_val_score(gbdt, train, target_train, cv=5)
       print(f"Threshold: {threshold}, Mean score: {np.mean(scores)}")
   

   • 在这个例子中，尝试了不同的阈值，并通过交叉验证评估了每个阈值下的模型性能。可以根据性能指标选择最佳的阈值。

需要注意的是，调整阈值是一个迭代的过程，需要根据具体的数据集和问题进行调整。同时，不同的阈值可能会对模型性能产生不同的影响，需要综合考虑特征数量和模型性能来选择合适的阈值。

原文地址：https://blog.csdn.net/rubyw/article/details/143087012

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！