From 1ce6856206097d7deb975567d17d853d5df91108 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 23 Feb 2024 15:31:26 +0100 Subject: SymbolicRegressionFunction: export hyper-parameters to dataref --- lib/functions.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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): -- cgit v1.2.3