sklearn里的决策树模型,DecisionTreeClassifier和DecisionTreeRegressor都可以设置最大深度max_depth。
我想知道在训练结束之后,怎么获得树的实际深度呢?
我觉得并不是每次训练树都能恰好达到最大深度,所以想看看实际深度。
1个回答
tree = DecisionTreeClassifier(max_depth=100)
tree.fit(X, y)
print(tree.tree_.max_depth)
赞,多谢多谢
-
jimmy
2019-03-27 22:26
相关讨论
sklearn里的RandomForestClassifier怎么选样本的数量
怎么得到sklearn里的RandomForestClassifier训练好之后每棵树的叶节点的个数?
sklearn里的RandomForestClassifier和ExtraTreesClassifier有什么区别吗?
sklearn模块里的DecisionTreeClassifier为什么每次结果不同(数据相同的情况下)
sklearn DecisionTreeRegressor模型中为什么会有随机种子random_state?
随便看看