functions: explicitly mention _xgb_ hyper-parameters in dref export

1 files 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,
         }