diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/functions.py | 13 |
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): |