diff --git a/autosklearn/automl.py b/autosklearn/automl.py index 85eb067ddf..6a56950394 100644 --- a/autosklearn/automl.py +++ b/autosklearn/automl.py @@ -406,7 +406,7 @@ def _fit(self, datamanager, metric, only_return_configuration_space=False): del self._datamanager except Exception: pass - + # => RUN SMAC smac_task_name = 'runSMAC' self._stopwatch.start_task(smac_task_name) @@ -581,6 +581,7 @@ def fit_ensemble(self, y, task=None, metric=None, precision='32', ensemble_nbest=ensemble_nbest, ensemble_size=ensemble_size) self._proc_ensemble.main() self._proc_ensemble = None + self._load_models() return self def _get_ensemble_process(self, time_left_for_ensembles,