作者回复: 很好的问题,你可以进行分类之后,查看下分类器中特征向量的重要程度,然后进行可视化的呈现,比如我们的分类器是clf,使用matplotlib进行可视化,具体代码如下:
# 显示特征向量的重要程度
coeffs = clf.feature_importances_
df_co = pd.DataFrame(coeffs, columns=["importance_"])
# 下标设置为Feature Name
df_co.index = train_features.columns
df_co.sort_values("importance_", ascending=True, inplace=True)
df_co.importance_.plot(kind="barh")
plt.title("Feature Importance")
plt.show()