画出lightGBM的特征重要性并排序

最新推荐文章于 2025-03-10 20:37:13 发布

码破苍穹

最新推荐文章于 2025-03-10 20:37:13 发布

阅读量4.5k

点赞数 1

文章标签：机器学习数据挖掘 python

本文链接：https://blog.csdn.net/leokingszx/article/details/122885911

版权

df = pd.DataFrame()
df['feature name'] = data_df.drop(["label"], axis=1).columns  #data_df为特征数据集
df['importance'] = clf.feature_importance()
df.sort_values(by='importance',inplace=True, ascending=False)

df2 = df.head(50)
df2.plot.barh(x = 'feature name',figsize=(10,12))