我创建了一个字典来匹配 sklearn 中决策树的特征重要性与我的 df 中相应的特征名称。这里是下面的代码:
importances = clf.feature_importances_
feature_names = ['age','BP','chol','maxh',
'oldpeak','slope','vessels',
'sex_0.0','sex_1.0',
'pain_1.0','pain_2.0','pain_3.0','pain_4.0',
'bs_0.0','bs_1.0',
'ecg_0.0','ecg_1.0','ecg_2.0',
'ang_0.0','ang_1.0',
'thal_3.0','thal_6.0','thal_7.0']
CLF_sorted = dict(zip(feature_names, importances))
在输出中我得到了这个:
{'BP': 0.053673644739136502,
'age': 0.014904980747733202,
'ang_0.0': 0.0,
'ang_1.0': 0.0,
'bs_0.0': 0.0,
'bs_1.0': 0.0,
'chol': 0.11125922817930389, ...}
正如我所料。我有两个问题要问你:
我怎样才能创建一个条形图,其中 x 轴代表
feature_names
y 轴对应的importances
?如果可能的话,我怎么能以降序对条形图进行排序?