Skip to content

Commit

Permalink
change 2-fold ridge default scoring function
Browse files Browse the repository at this point in the history
* change it to mean squared error as in RidgeCV
  • Loading branch information
agoscinski committed Aug 7, 2023
1 parent 8acd86d commit adeaac6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/skmatter/linear_model/_ridge.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit adeaac6

Please sign in to comment.