对sklearn的决策树进行剪枝

  统计/机器学习 监督式学习 Python    浏览次数:3917        分享
2

请教各位,可以对sklearn的决策树模型进行剪枝吗?

 

张球球   2017-10-13 09:37



   3个回答 
4

目前还没有。

不过。

在山的那边,海的那边,有一群勤劳的程序员,他们正在讨论在新版本的sklearn里增加post-prune。


链接在此

https://github.com/scikit-learn/scikit-learn/issues/6557

SofaSofa数据科学社区DS面试题库 DS面经

Lydia   2017-10-13 12:17

2

sklearn目前没有具体实现剪枝的功能。

现在能做的是预剪枝,就是设置Classifier或者Regression里的参数max_depth, min_samples_split, min_samples_leaf。

后剪枝的确是在sklearn中做不到的。


----------补充-----------

什么是预剪枝,后剪枝?

SofaSofa数据科学社区DS面试题库 DS面经

派大星   2017-10-13 10:14

2

当然有,这个是针对sklearn的决策树进行剪枝的完整代码和解析

https://blog.csdn.net/appleyuchi/article/details/83692381

SofaSofa数据科学社区DS面试题库 DS面经

appleyuchi   2019-07-11 11:35



  相关主题

可视化sklearn决策树的结构   2回答

sklearn里怎么获得adaboost中每棵树的权重?   1回答

sklearn随机森林设置每棵树的权重   2回答

怎么获得sklearn里决策树的实际深度?   1回答

sklearn里的RandomForestClassifier和ExtraTreesClassifier有什么区别吗?   1回答

sklearn模块里的DecisionTreeClassifier为什么每次结果不同(数据相同的情况下)   1回答

sklearn里的RandomForestClassifier怎么选样本的数量   1回答

python sklean中的决策树是用的哪一种决策树算法?   2回答

怎么得到randomforestregressor里每棵树的预测结果?   1回答

Adaboost里的树有没有可能权重是负数?   1回答

Extra Tree算法   1回答

随机森林中增加树的数量对于偏差和方差的影响   2回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!