# 特征重要性
import matplotlib.pyplot as plt
import seaborn as sns
cols = (feature_importance_data[["feature", "importance"]]
.groupby("feature")
.mean()
.sort_values(by="importance", ascending=False)[:1000].index)
best_features = feature_importance_data.loc[feature_importance_data.feature.isin(cols)]
plt.figure(figsize=(14,40))
sns.barplot(x="importance",
y="feature",
data=best_features.sort_values(by="importance",
ascending=False))
plt.title('LightGBM Features (avg over folds)')
plt.tight_layout()
plt.show()
数据挖掘-feature_importanct
最新推荐文章于 2021-12-07 22:03:53 发布