获取特征名称

Getting Feature Names

假设我有 4 个特征(名称)['year2000'、'year2001'、'year2002'、'year2003'],用于决策树分类器的学习。 我如何从 feature_importances_since 中获取重要特征的名称它直接给我一些数字而不是特征名称

假设您将功能名称放入列表中

feature_names = ['year2000', 'year2001','year2002','year2003']

那么问题就是获取重要性最高的特征的索引

feature_importances = clf.feature_importances_
k = 3
top_k_idx = feature_importances.argsort()[-k:][::-1]
print feature_names[top_k_idx]