Skip to content

Commit 446b7d6

Browse files
committed
Move resolve ensemble class check to init
1 parent 04aa850 commit 446b7d6

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

autosklearn/estimators.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,12 @@ def __init__(
405405
)
406406

407407
# Need to resolve the ensemble class here so we can act on it below.
408-
ensemble_class = self.resolve_ensemble_class(ensemble_class, metric)
408+
if ensemble_class == "default":
409+
ensemble_class = (
410+
MultiObjectiveDummyEnsemble
411+
if isinstance(metric, Scorer) and len(metric) > 1
412+
else EnsembleSelection
413+
)
409414
self.ensemble_class = ensemble_class
410415

411416
# User specified `ensemble_size` explicitly, warn them about deprecation
@@ -742,15 +747,6 @@ def fit_ensemble(
742747
)
743748
return self
744749

745-
def resolve_ensemble_class(self, ensemble_class, metric):
746-
if ensemble_class == "default":
747-
ensemble_class = (
748-
EnsembleSelection
749-
if metric is None or isinstance(metric, Scorer) or len(metric) == 1
750-
else MultiObjectiveDummyEnsemble
751-
)
752-
return ensemble_class
753-
754750
def refit(self, X, y):
755751
"""Refit all models found with fit to new data.
756752

0 commit comments

Comments
 (0)