From 2e29ebaca6bc26fa838f7c3b8b13960c600884e4 Mon Sep 17 00:00:00 2001 From: Matthias Feurer Date: Fri, 24 Feb 2017 16:57:18 +0100 Subject: [PATCH] FIX estimator signature for regression --- autosklearn/estimators.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/autosklearn/estimators.py b/autosklearn/estimators.py index 278fdca6e0..1066431bc4 100644 --- a/autosklearn/estimators.py +++ b/autosklearn/estimators.py @@ -434,7 +434,9 @@ def fit(self, X, y, """ # Fit is supposed to be idempotent! # But not if we use share_mode. - return super(AutoSklearnRegressor, self).fit(X, y, metric, feat_type, dataset_name) + return super(AutoSklearnRegressor, self).fit(X=X, y=y, metric=metric, + feat_type=feat_type, + dataset_name=dataset_name) def predict(self, X): """Predict regression target for X. @@ -559,5 +561,4 @@ def fit(self, X, y, dataset_name=None, ): return self._automl.fit(X=X, y=y, task=REGRESSION, metric=metric, - loss=loss, feat_type=feat_type, - dataset_name=dataset_name) + feat_type=feat_type, dataset_name=dataset_name)