请教各位,可以对sklearn的决策树模型进行剪枝吗?
3个回答
目前还没有。
不过。
在山的那边,海的那边,有一群勤劳的程序员,他们正在讨论在新版本的sklearn里增加post-prune。
链接在此
https://github.com/scikit-learn/scikit-learn/issues/6557
SofaSofa数据科学社区DS面试题库 DS面经sklearn目前没有具体实现剪枝的功能。
现在能做的是预剪枝,就是设置Classifier或者Regression里的参数max_depth, min_samples_split, min_samples_leaf。
后剪枝的确是在sklearn中做不到的。
----------补充-----------
SofaSofa数据科学社区DS面试题库 DS面经当然有,这个是针对sklearn的决策树进行剪枝的完整代码和解析
https://blog.csdn.net/appleyuchi/article/details/83692381
相关讨论
sklearn DecisionTreeRegressor模型中为什么会有随机种子random_state?
怎么得到sklearn里的RandomForestClassifier训练好之后每棵树的叶节点的个数?
sklearn里的RandomForestClassifier怎么选样本的数量
lightgbm.LGBMModel和lightgbm.LGBMClassifier该用哪个?
sklearn里的RandomForestClassifier和ExtraTreesClassifier有什么区别吗?
随便看看