Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs
Apache License 2.0
692 stars 87 forks source link

Is there a parameter to set "minimum samples per leaf" for Random Forest? #70

Closed DonSeger closed 2 years ago

DonSeger commented 2 years ago

I wonder if there is a parameter that corresponds to "minimum samples per leaf" for the Random Forest? If yes, which one? If no, what is the default?

I have seen the page that describes ThunderGBM Parameters, but couldn't find "minimum samples per leaf". https://github.com/Xtra-Computing/thundergbm/blob/master/docs/parameters.md

zeyiwen commented 2 years ago

min_child_weight as described in the Parameters page.