特征重要度

最新推荐文章于 2024-01-18 10:17:04 发布

xiedelong

最新推荐文章于 2024-01-18 10:17:04 发布

阅读量634

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/xiedelong/article/details/109053665

版权

算法专栏收录该内容

71 篇文章 4 订阅

订阅专栏

# 特征重要性
feat_imp_split = pd.DataFrame({
        'column': list(x_train.columns),
        'importance': clf.feature_importance(importance_type='split'), # 'importance': clf.feature_importance(importance_type='gain'),
    }).sort_values(by='importance', ascending=True)

feat_imp_split.plot(x='column', y='importance', kind='barh',figsize=(12.5,50),title='Feature Importances')
plt.xlabel('Feature Importance Score')