获取功能名称

时间:2015-07-01 08:05:47

标签: scikit-learn

假设我有4个功能(名称)s ['year2000','year2001','year2002','year2003'],在学习决策树分类器时使用。 如何从feature_importances_s获取重要特征的名称,因为它直接给我一些数字而不是特征名称

1 个答案:

答案 0 :(得分:1)

假设您将要素名称放在列表中

<?php
print_r(glob("/files/*.pdf"));
?> 

然后问题只是获得具有最高k重要性的特征索引

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