特征重要度

# 特征重要性
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')

特征重要度_第1张图片

你可能感兴趣的:(算法)