sklearn里的RandomForestClassifier训练好之后每棵树的结构都不大一样，比如叶节点的个数很可能不同。
怎么获得随机森林中每棵树的叶节点的个数？
1个回答
如果sklearn的版本在0.21以上的话，下面的代码可以直接用
from sklearn.ensemble import RandomForestClassifier
clf = RandomForestClassifier(n_estimators =5)
clf.fit(x, y)
num_leaves_each_tree = [tree.get_n_leaves() for tree in clf.estimators_]
matt
2020-06-30 20:50
相关讨论
随便看看