SymbolicRegressionFunction: export hyper-parameters to dataref

1 files changed, 13 insertions, 0 deletions

diff --git a/lib/functions.py b/lib/functions.py
index c6ea283..f557cbb 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -1278,6 +1278,19 @@ class SymbolicRegressionFunction(SKLearnRegressionFunction):
         rstr = str(self.regressor)
         return rstr.count(",") * 2 + 1
 
+    def hyper_to_dref(self):
+        return {
+            "symreg/population size": self.regressor.population_size,
+            "symreg/generations": self.regressor.generations,
+            "symreg/tournament size": self.regressor.tournament_size,
+            "symreg/const range/min": self.regressor.const_range[0],
+            "symreg/const range/max": self.regressor.const_range[1],
+            "symreg/function set": " ".join(self.regressor.function_set),
+            "symreg/metric": self.regressor.metric,
+            "symreg/parsimony coefficient": self.regressor.parsimony_coefficient,
+            "symreg/n jobs": self.regressor.n_jobs,
+        }
+
 
 # first-order linear function (no feature interaction)
 class FOLFunction(ModelFunction):