Skip to content

Commit 63b8f73

Browse files
adding fit_params as kwargs
1 parent 62e5866 commit 63b8f73

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

ISLP/models/sklearn_wrap.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def __sklearn_tags__(self):
6060
tags.estimator_type = 'classifier'
6161
return tags
6262

63-
def fit(self, X, y):
63+
def fit(self, X, y, **fit_params):
6464
"""
6565
Fit a statsmodel model
6666
with design matrix
@@ -185,7 +185,7 @@ def __sklearn_tags__(self):
185185
tags = super().__sklearn_tags__()
186186
return tags
187187

188-
def fit(self, X, y):
188+
def fit(self, X, y, **fit_params):
189189
"""
190190
First, select a model
191191
with design matrix
@@ -275,7 +275,7 @@ def __init__(self,
275275
self.cv = cv
276276
self.scoring = scoring
277277

278-
def fit(self, X, y):
278+
def fit(self, X, y, **fit_params):
279279

280280
"""
281281
First, select a model

0 commit comments

Comments
 (0)