Skip to content

Commit 40f78ad

Browse files
authored
Add kwargs to AutoMLRegressor init
`AutoMLRegressor` is currently lacking an `__init___()` method to be able to override default args to `AutoML` base class. This pull requests adds this functionality based on the existing code for `AutoMLClassifier`.
1 parent f39c8d3 commit 40f78ad

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

autosklearn/automl.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,6 +1057,9 @@ def predict_proba(self, X, batch_size=None, n_jobs=1):
10571057

10581058

10591059
class AutoMLRegressor(BaseAutoML):
1060+
def __init__(self, *args, **kwargs):
1061+
super().__init__(*args, **kwargs)
1062+
10601063
def fit(
10611064
self,
10621065
X: np.ndarray,

0 commit comments

Comments
 (0)