Как искать особенности случайного дерева дерева/дерева решений?

Question

Я использую python для создания классификатора дерева решений. Я хочу видеть функции NAME, которые я использовал из класса. Вот код

In [3]: my_forest.feature_importances_ 
Out[3]: 
array([ 0.09546647, 0.210264 , 0.31768782, 0.24191602, 0.05580487, 
     0.04318897, 0.03567185]) 

Я могу найти 7 функций, но имя не отображается.

Answer 1

преобразование данных в словарь с помощью:

dv = DictVectorizer() 
newX = dv.fit_transform(all_data) 

Теперь используйте newX вместо X для обучения и тестирования. Затем вы можете сделать

tree.export_graphviz(classifier_used, feature_names=dv.get_feature_names(), out_file='graph.dot') 

Это поможет вам визуализировать все функции.