From c680e131ceb8403031f4f2efe8b67548f5c0be06 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 19 Jan 2024 10:46:20 +0100 Subject: functions: explicitly mention _xgb_ hyper-parameters in dref export --- lib/functions.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/lib/functions.py b/lib/functions.py index 82e516e..0f472ee 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -733,13 +733,13 @@ class XGBoostFunction(SKLearnRegressionFunction): def to_dref(self): return { - "hyper/n estimators": self.regressor.n_estimators, - "hyper/max depth": self.regressor.max_depth, - "hyper/subsample": self.regressor.subsample, - "hyper/eta": self.regressor.learning_rate, - "hyper/gamma": self.regressor.gamma, - "hyper/alpha": self.regressor.reg_alpha, - "hyper/lambda": self.regressor.reg_lambda, + "xgb hyper/n estimators": self.regressor.n_estimators, + "xgb hyper/max depth": self.regressor.max_depth, + "xgb hyper/subsample": self.regressor.subsample, + "xgb hyper/eta": self.regressor.learning_rate, + "xgb hyper/gamma": self.regressor.gamma, + "xgb hyper/alpha": self.regressor.reg_alpha, + "xgb hyper/lambda": self.regressor.reg_lambda, } -- cgit v1.2.3