diff options
Diffstat (limited to 'lib/functions.py')
| -rw-r--r-- | lib/functions.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/lib/functions.py b/lib/functions.py index 32f777b..d18477e 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -763,7 +763,12 @@ class XGBoostFunction(SKLearnRegressionFunction): def hyper_to_dref(self): return { "xgb/n estimators": self.regressor.n_estimators, - "xgb/max depth": self.regressor.max_depth, + "xgb/max depth": self.regressor.max_depth == 0 + and "infty" + or self.regressor.max_depth, + "xgb/max leaves": self.regressor.max_leaves == 0 + and "infty" + or self.regressor.max_leaves, "xgb/subsample": self.regressor.subsample, "xgb/eta": self.regressor.learning_rate, "xgb/gamma": self.regressor.gamma, |