Как вы получаете доступ к глубине дерева в Python's scikit-learn?

Question

Я использую scikit-learn для создания случайного леса. Однако я хочу найти индивидуальные глубины каждого дерева. Это похоже на простой атрибут, но в соответствии с документацией (http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html) доступ к нему отсутствует.

Если это невозможно, есть ли способ доступа к глубине дерева из дерева дерева решений?

Любая помощь будет оценена по достоинству. Спасибо.

Answer 1

Каждый экземпляр RandomForestClassifier имеет атрибут estimators_, который представляет собой список DecisionTreeClassifier экземпляров. Документация показывает, что экземпляр DecisionTreeClassifier имеет атрибут tree_, который является экземпляром класса (недокументированный, я считаю) Tree. Некоторые исследования в интерпретаторе показывают, что каждый экземпляр Tree имеет параметр max_depth, который появляется, чтобы быть тем, что вы ищете - опять же, это недокументировано.

В любом случае, если forest ваш экземпляр RandomForestClassifier, то:

>>> [estimator.tree_.max_depth for estimator in forest.estimators_] 
[9, 10, 9, 11, 9, 9, 11, 7, 13, 10] 

должен сделать трюк.