Skip to content

Click in-app to access the full platform documentation for your version of DataRobot.

Optimal binning in DataRobot

Robot 1

Got a question from a client—does DataRobot help do optimal binning (on numeric features)? I know that in our GAM model, DataRobot will bin each of the numeric features based on its partial dependence from an XGBoost model. Anything else that we do that I am not aware of?

Robot 2

We use a decision tree to find bins.

It's pretty optimal.

It may not be perfectly optimal, but it does a good job finding the right bins.

Robot 3

Single decision tree on single feature—produces leaves with at least minimum number of target values. So bins are variable size and are designed to have enough target statistics per leaf. The boundaries are just sorted splits. An XGBoost model is used to smooth the target and decision tree operate on XGB predictions.

Updated September 18, 2023
Back to top