scikit-learn: Итерация по узлам DecisionTreeClassifier

Question

12 from sklearn.datasets import load_iris 
13 iris = load_iris() 
14 X = iris.data 
15 y = iris.target 
16 
19 clf = DecisionTreeClassifier() 
20 clf = clf.fit(iris.data,iris.target) 

Как я могу перебирать узлы clf. Я не мог найти его нигде в документации.

Answer 1

Что вы хотите делать с узлами clf?

Существует переменная, называемая clf.tree_, которая содержит фактическую информацию дерева решений. Он плохо документирован в пользовательских документах, но вы можете read the code, чтобы лучше понять, что он делает.

К сожалению, фактический массив узла, кажется, скрыт в атрибуте Cython, но вы можете использовать целочисленные индексы 0...clf.tree_.node_count как индексы в clf.tree_.feature[i], clf.tree_.threshold[i] и т.д. (см документации в связанной коде для получения дополнительной информации). Если вы хотите выяснить, на каком узле находится образец, вы можете использовать clf.tree_.apply(X) для получения фактического целочисленного индекса узла.