File tree Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments