Skip to content

Commit 2e29eba

Browse files
committed
FIX estimator signature for regression
1 parent f5608e2 commit 2e29eba

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

autosklearn/estimators.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -434,7 +434,9 @@ def fit(self, X, y,
434434
"""
435435
# Fit is supposed to be idempotent!
436436
# But not if we use share_mode.
437-
return super(AutoSklearnRegressor, self).fit(X, y, metric, feat_type, dataset_name)
437+
return super(AutoSklearnRegressor, self).fit(X=X, y=y, metric=metric,
438+
feat_type=feat_type,
439+
dataset_name=dataset_name)
438440

439441
def predict(self, X):
440442
"""Predict regression target for X.
@@ -559,5 +561,4 @@ def fit(self, X, y,
559561
dataset_name=None,
560562
):
561563
return self._automl.fit(X=X, y=y, task=REGRESSION, metric=metric,
562-
loss=loss, feat_type=feat_type,
563-
dataset_name=dataset_name)
564+
feat_type=feat_type, dataset_name=dataset_name)

0 commit comments

Comments
 (0)