LightGBM.feature_importance()

最新推荐文章于 2024-10-04 17:45:00 发布

HuJH5145

最新推荐文章于 2024-10-04 17:45:00 发布

阅读量8.9k

点赞数 3

分类专栏：毕业设计 Python

本文链接：https://blog.csdn.net/qq_41904729/article/details/117928981

版权

毕业设计同时被 2 个专栏收录

28 篇文章 1 订阅

订阅专栏

Python

22 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用LightGBM库的feature_importance()函数对训练好的模型特征进行重要性评估，并通过Seaborn绘制条形图进行可视化展示。展示了特征如'file_id_api_nunique'、'file_id_api_count'等的重要性排序，以及如何设置图表参数以提高可读性。

摘要由CSDN通过智能技术生成

使用LightGBM.feature_importance()函数给训练完毕的LightGBM模型的各特征进行重要性排序。

feature_importance             = pd.DataFrame()
feature_importance['fea_name'] = train_features
feature_importance['fea_imp']  = clf.feature_importance()
feature_importance             = feature_importance.sort_values('fea_imp',ascending = False)

plt.figure(figsize=[20,10],dpi=100)
ax = sns.barplot(x = feature_importance['fea_name'], y = feature_importance['fea_imp'])
ax.set_xticklabels(labels = ['file_id_api_nunique','file_id_api_count','file_id_tid_max','file_id_tid_mean','file_id_tid_min','file_id_tid_std','file_id_index_mean','file_id_tid_nunique','file_id_index_nunique','file_id_index_std','file_id_index_max','file_id_tid_count','file_id_index_count','file_id_index_min'],
                                    rotation = 45,fontsize = 15)
ax.set_yticklabels(labels = [0,2000,4000,6000,8000,10000,12000,14000,16000],fontsize = 15)
plt.xlabel('fea_name',fontsize=18)
plt.ylabel('fea_imp',fontsize=18)
# plt.tight_layout()
plt.savefig('D:/A_graduation_project/pictures/2_baseline1/特征重要性')

官方文档

feature_importance(importance_type='split', iteration=-1)

Get feature importances.

Parameters:
- importance_type (string__, optional (default="split")) – How the importance is calculated. If “split”, result contains numbers of times the feature is used in a model. If “gain”, result contains total gains of splits which use the feature.
- iteration (int or None, optional (default=None)) – Limit number of iterations in the feature importance calculation. If None, if the best iteration exists, it is used; otherwise, all trees are used. If <= 0, all trees are used(no limits).
Returns:
- result – Array with feature importances.
Return type:
- numpy array