Skip to content

Commit adeaac6

Browse files
committed
change 2-fold ridge default scoring function
* change it to mean squared error as in RidgeCV
1 parent 8acd86d commit adeaac6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/skmatter/linear_model/_ridge.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ class RidgeRegression2FoldCV(BaseEstimator, MultiOutputMixin, RegressorMixin):
8787
A string (see model evaluation documentation) or
8888
a scorer callable object / function with signature
8989
``scorer(estimator, X, y)``.
90-
If None, the negative root mean squared error is used.
90+
If None, the negative mean squared error is used.
9191
n_jobs : int, default=None
9292
The number of CPUs to use to do the computation.
9393
:obj:`None` means 1 unless in a :obj:`joblib.parallel_backend` context.
@@ -166,7 +166,7 @@ def fit(self, X, y):
166166
# intercepted here
167167
if self.scoring is None:
168168
scorer = check_scoring(
169-
self, scoring="neg_root_mean_squared_error", allow_none=False
169+
self, scoring="neg_mean_squared_error", allow_none=False
170170
)
171171
else:
172172
scorer = check_scoring(self, scoring=self.scoring, allow_none=False)

0 commit comments

Comments
 (0)