embed icon indicating copy to clipboard operation
embed copied to clipboard

User defined cuts for tree-based methods

Open wbuchanan opened this issue 2 years ago • 0 comments

Feature

It isn't currently clear how step_collapse_cart and other methods for numeric predictors determine how many groups to select. It could be useful to add a parameter cut_depth that would allow users to specify how far down the decision tree to make cuts to collapse/group values. Alternatively, a groups parameter that allows users to specify the number of groups they want to retain, which would return <= groups based on cutting the tree could also be a useful feature for users.

wbuchanan avatar Sep 20 '23 05:09 wbuchanan