soporte - ¿Cómo se accede a la profundidad del árbol en el scikit-learn de Python?

python machine learning pdf

Question

Estoy usando scikit-learn para crear un Random Forest. Sin embargo, quiero encontrar las profundidades individuales de cada árbol. Parece un atributo simple pero según la documentación ( http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html ) no hay forma de acceder a él.

Si esto no es posible, ¿hay alguna manera de acceder a la profundidad del árbol desde un modelo de Decision Tree?

Cualquier ayuda sería apreciada. Gracias.

Answer 1

Cada instancia de RandomForestClassifier tiene un atributo estimators_ , que es una lista de instancias de DecisionTreeClassifier . La documentación muestra que una instancia de DecisionTreeClassifier tiene un atributo tree_ , que es una instancia de la clase de Tree (indocumentada, creo). Algunas exploraciones en el intérprete muestran que cada instancia de Tree tiene un parámetro max_depth que parece ser lo que está buscando; de nuevo, está indocumentado.

En cualquier caso, si forest es tu instancia de RandomForestClassifier , entonces:

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

debería hacer el truco.