我已经创建了一个随机森林分类器，并且正在尝试生成我的随机森林模型中树深度的直方图。但是我无法提取森林中每棵树的深度。

我的RF模型名为’RF_optimised’，我尝试过下面的代码来迭代我的树并进行可视化，这个方法是有效的。我查看了estimators_和export_graphviz的文档，但似乎没有办法提取树的实际深度。

from sklearn import treefrom sklearn.tree import export_graphvizfrom sklearn.externals.six import StringIO# 创建一个字符串缓冲区来写入（一个假的文本文件）f = StringIO()i_tree = 0for tree_in_forest in RF_optimised.estimators_:    export_graphviz(tree_in_forest,out_file=f,    #feature_names=col,    filled=True,    rounded=True,    proportion=True)    graph = pydotplus.graph_from_dot_data(f.getvalue())    display(Image(graph.create_png()))

我需要一个函数来迭代我的随机森林中的树，并将树的深度存储在一个列表或数据框中，以便稍后生成直方图。谁能帮帮我？

回答：

在解释器中的一些探索显示，每个Tree实例都有一个max_depth参数，看起来这就是我所寻找的——再次强调，这没有在文档中说明。

[estimator.tree_.max_depth for estimator in RF_optimised.estimators_]

对我来说这个方法奏效了 🙂