我正在使用scikit-learn来创建一个随机森林.但是,我想找到每棵树的各个深度.这似乎是一个简单的属性,但根据文档,(http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html),无法访问它.
如果这不可能,有没有办法从决策树模型访问树深度?
任何帮助,将不胜感激.谢谢.
每个实例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]
应该做的伎俩.