3
$\begingroup$

Why pruning is not currently supported in scikit-learn? How can we tune the decision trees to make a workaround?

Improve this question
$\endgroup$

3 Answers 3

Reset to default
5
$\begingroup$

This is something which is planned to be done. Setting the minimum number of samples required at a leaf node or a split as well as setting the maximum depth of the tree are how you want to work around this.

Improve this answer
$\endgroup$
0
2
$\begingroup$

I have performed Cost Complexity Pruning Algorithm on Sklearn's CART(Classification) model,

I have performed Error Complexity Pruning Algorithm on Sklearn's CART(Regression) model.

It is available here.

A relevant issue for sklearn about your question is here.

Improve this answer
$\endgroup$
1
$\begingroup$

Have a look at the 0.22 dev version of sklearn. Looks like tree pruning will be implemented in the next version.

https://scikit-learn.org/dev/whats_new.html section sklearn.tree

Improve this answer
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.