From adeaac6a3541768289cff0228beb52021f80c118 Mon Sep 17 00:00:00 2001 From: Alexander Goscinski Date: Mon, 31 Jul 2023 22:54:21 +0200 Subject: [PATCH] change 2-fold ridge default scoring function * change it to mean squared error as in RidgeCV --- src/skmatter/linear_model/_ridge.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/skmatter/linear_model/_ridge.py b/src/skmatter/linear_model/_ridge.py index f34cf3606..0d769e561 100644 --- a/src/skmatter/linear_model/_ridge.py +++ b/src/skmatter/linear_model/_ridge.py @@ -87,7 +87,7 @@ class RidgeRegression2FoldCV(BaseEstimator, MultiOutputMixin, RegressorMixin): A string (see model evaluation documentation) or a scorer callable object / function with signature ``scorer(estimator, X, y)``. - If None, the negative root mean squared error is used. + If None, the negative mean squared error is used. n_jobs : int, default=None The number of CPUs to use to do the computation. :obj:`None` means 1 unless in a :obj:`joblib.parallel_backend` context. @@ -166,7 +166,7 @@ def fit(self, X, y): # intercepted here if self.scoring is None: scorer = check_scoring( - self, scoring="neg_root_mean_squared_error", allow_none=False + self, scoring="neg_mean_squared_error", allow_none=False ) else: scorer = check_scoring(self, scoring=self.scoring, allow_none=False)