From a808b61d2d4ca223c5195882979089e35c482936 Mon Sep 17 00:00:00 2001 From: chico Date: Thu, 1 Apr 2021 16:49:20 +0200 Subject: [PATCH] bug_fix_show_models --- autosklearn/evaluation/abstract_evaluator.py | 6 ++++++ test/test_automl/test_automl.py | 11 +++++++++++ 2 files changed, 17 insertions(+) diff --git a/autosklearn/evaluation/abstract_evaluator.py b/autosklearn/evaluation/abstract_evaluator.py index 8178688e58..50a9cd272e 100644 --- a/autosklearn/evaluation/abstract_evaluator.py +++ b/autosklearn/evaluation/abstract_evaluator.py @@ -57,6 +57,9 @@ def __init__( super(MyDummyClassifier, self).__init__(strategy="most_frequent") self.random_state = random_state self.init_params = init_params + self.dataset_properties = dataset_properties + self.include = include + self.exclude = exclude def pre_transform( self, @@ -110,6 +113,9 @@ def __init__( super(MyDummyRegressor, self).__init__(strategy='median') self.random_state = random_state self.init_params = init_params + self.dataset_properties = dataset_properties + self.include = include + self.exclude = exclude def pre_transform( self, diff --git a/test/test_automl/test_automl.py b/test/test_automl/test_automl.py index c358ac9553..71da849358 100644 --- a/test/test_automl/test_automl.py +++ b/test/test_automl/test_automl.py @@ -12,6 +12,7 @@ import pandas as pd import pytest import sklearn.datasets +from sklearn.base import clone from smac.scenario.scenario import Scenario from smac.facade.roar_facade import ROAR @@ -426,6 +427,16 @@ def test_do_dummy_prediction(backend, dask_client, datasets): 'predictions_ensemble_1_1_0.0.npy') ) + model_path = os.path.join(backend.temporary_directory, '.auto-sklearn', + 'runs', '1_1_0.0', + '1.1.0.0.model') + + # Make sure the dummy model complies with scikit learn + # get/set params + assert os.path.exists(model_path) + with open(model_path, 'rb') as model_handler: + clone(pickle.load(model_handler)) + auto._clean_logger() del auto