Python机器学习：plot_importance()查看特征重要度

lightgmb算法里面的plot_importance()方法支持特征重要度的查看，下面将以lightgmb算法为例将特征重要度可视化展示出来。另外xgboost算法的实现也几乎一样哦。

事先准备好模型：

import lightgbm as lgb
model_lgb = lgb.LGBMClassifier().fit(X_train, y_train)

以上模型训练好了，下面查看特征重要度：

from lightgbm import plot_importance
fig,ax = plt.subplots(figsize=(10,8))
plot_importance(model_lgb,max_num_features=20,ax=ax)
plt.show()

代码讲解：

import导入lightgbm算法里查看特征重要度的plot_importance包；

plt.subplots(figsize=(10,8))指生成长为10，宽为8的画布；

plot_importance()里面的model_lgb是我们事先定义的函数名，里面存了lightgbm算法；max_num_features=20展示头部20个特征；

运行结果：

不限制max_num_features，即可展示所有的特征：

fig,ax = plt.subplots(figsize=(10,8))
plot_importance(model_lgb,ax=ax)

图里面的特征重要度从大到小的排列，就能够直观地了解哪些是影响预测结果的重要特征了。

Python机器学习：plot_importance()查看特征重要度相关推荐

R语言使用randomForest包构建随机森林模型（Random forests）、使用importance函数查看特征重要度、使用table函数计算混淆矩阵评估分类模型性能、包外错误估计OOB
R语言使用randomForest包中的randomForest函数构建随机森林模型(Random forests).使用importance函数查看特征重要度.使用table函数计算混淆矩阵评估分类 ...
杨桃的Python机器学习3——单特征与多特征、训练集与测试集
本人CSDN博客专栏:https://blog.csdn.net/yty_7 Github地址:https://github.com/yot777/ 单特征与多特征在上一节标签和特征的示例中,我们使 ...
python 机器学习管道_构建机器学习管道-第1部分
python 机器学习管道 Below are the usual steps involved in building the ML pipeline: 以下是构建ML管道所涉及的通常步骤: Imp ...
python机器学习预测_使用Python和机器学习预测未来的股市趋势
python机器学习预测 Note from Towards Data Science's editors: While we allow independent authors to publish ...
Python机器学习入门笔记（1）—— Scikit-learn与特征工程
目录机器学习算法分类数据集工具 Scikit-learn Scikit-learn的安装 scikit-learn数据集API介绍 bunch对象 datasets模块数据集的划分 train_ ...
Python机器学习—特征工程
文章目录 1.数据集 1.1 可用数据集 1.2 scikit-learn数据集 sklearn小数据集 sklearn大数据集 1.3 数据集的划分数据集划分API 2.特征工程 2.1特征工程包 ...
【机器学习】用特征量重要度(feature importance)解释模型靠谱么？怎么才能算出更靠谱的重要度？
[机器学习]用特征量重要度(feature importance)解释模型靠谱么?怎么才能算出更靠谱的重要度? 我们用机器学习解决商业问题的时候,不仅需要训练一个高精度高泛化性的模型,往往还需要解释哪 ...
Python 随机森林特征重要度
Python 随机森林特征重要度 1 声明本文的数据来自网络,部分代码也有所参照,这里做了注释和延伸,旨在技术交流,如有冒犯之处请联系博主及时处理. 2 随机森林特征重要度简介决策树的优点是通过树 ...
Python机器学习建模的标准流程（特征工程+建模调参+模型评估+全流程可视化）
作者CSDN:进击的西西弗斯本文链接:https://blog.csdn.net/qq_42216093/article/details/116994199 版权声明:本文为作者原创文章,转载需征得 ...

Python机器学习：plot_importance()查看特征重要度

Python机器学习：plot_importance()查看特征重要度相关推荐

最新文章

热门文章