diff options
| -rw-r--r-- | lib/functions.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/functions.py b/lib/functions.py index 07d3e2f..fdd2cfb 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -1193,9 +1193,11 @@ class SymbolicRegressionFunction(SKLearnRegressionFunction): from dfatool.gplearn.genetic import SymbolicRegressor - self.regressor = SymbolicRegressor() - self.regressor.fit(fit_parameters, data) self._build_feature_names() + self.regressor = SymbolicRegressor( + metric="mse", feature_names=self.feature_names + ) + self.regressor.fit(fit_parameters, data) self.fit_success = True return self |