如何显示模型特征权重占比图【数据分析】

可视化模型的特征权重

1、流程

1、导入库:
numpy:用于处理数组和矩阵。
matplotlib.pyplot:用于绘图。
sklearn.datasets:用于加载数据集。
sklearn.ensemble.RandomForestClassifier:用于训练随机森林模型。

2、加载数据集:
使用load_iris函数加载Iris数据集。

3、训练模型:
初始化RandomForestClassifier并训练模型。

4、提取特征重要性:
使用model.feature_importances_获取特征重要性。

5、绘制条形图:
使用plt.barh绘制水平条形图。
plt.yticks设置y轴标签为特征名称。
plt.gca().invert_yaxis()反转y轴,使最重要的特征显示在图的顶部。

2、feature_importances_获取特征重要性

以鸢尾花为例,有以下特征:

'sepal length (cm)':萼片长度(厘米)
萼片是花朵最外层的绿色或彩色的叶状结构,这个特征指的是萼片的长度,单位是厘米。
'sepal width (cm)':萼片宽度(厘米)
这个特征指的是萼片的宽度,单位也是厘米。
'petal length (cm)':花瓣长度(厘米)
花瓣是花朵中色彩鲜艳、常常用于吸引传粉者的部分,这个特征指的是花瓣的长度,单位是厘米。
'petal width (cm)':花瓣宽度(厘米)
这个特征指的是花瓣的宽度,单位同样是厘米。
</
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

东木月

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值