From bb773703ffa9951e31acb21c09e9b1fd151836fa Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Wed, 21 Feb 2024 15:24:33 +0100 Subject: SymReg: use MSE and set feature_names --- lib/functions.py | 6 ++++-- 1 file 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 -- cgit v1.2.3