lightgbm 重要性画图

lightgbm本身自带plot_importances方法,但是画出来不怎么美观,似乎一般是把importance数据转成dataframe,然后再进行画图

feature_imp = pd.DataFrame(sorted(zip(model.feature_importance(importance_type='gain'),X.columns)), columns=['Value','Feature'])
plt.figure(figsize=(12, 20))
sns.barplot(x="Value", y="Feature", data=feature_imp.sort_values(by="Value", ascending=False).iloc[:20,:])
plt.title('LightGBM Features')
plt.show()
以上输出前20重要的变量


留言

熱門文章