首页 > 解决方案 > 如何在特征重要性条形图上显示特征名称?

问题描述

我使用了特征选择,然后使用了分类器模型。然而,当使用特征重要性绘制条形图时,我看不到特征名称。我应该怎么办?

# Removing features with low variance
from sklearn.feature_selection import VarianceThreshold
selector = VarianceThreshold(threshold=(.8 * (1 - .8)))
selector.fit(X_train)
X_train = selector.transform(X_train)
X_test = selector.transform(X_test)
    
# Create a dataframe of feature importance 
df_feature_importance = pd.DataFrame(model_CBC.get_feature_importance(prettified=True,)[1:10])
#plotting feature importance
plt.figure(figsize=(8, 5));
feature_plot= sns.barplot(x="Importances", y="Feature Id", data=df_feature_importance,palette="icefire");
plt.title('features importance')

结果: 结果

标签: machine-learningscikit-learnbar-chartfeature-selection

解决方案


推荐阅读