Skip to content

Commit f6d6be4

Browse files
bug_fix_show_models (#1114)
1 parent 4d3bb06 commit f6d6be4

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

autosklearn/evaluation/abstract_evaluator.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,9 @@ def __init__(
5757
super(MyDummyClassifier, self).__init__(strategy="most_frequent")
5858
self.random_state = random_state
5959
self.init_params = init_params
60+
self.dataset_properties = dataset_properties
61+
self.include = include
62+
self.exclude = exclude
6063

6164
def pre_transform(
6265
self,
@@ -110,6 +113,9 @@ def __init__(
110113
super(MyDummyRegressor, self).__init__(strategy='median')
111114
self.random_state = random_state
112115
self.init_params = init_params
116+
self.dataset_properties = dataset_properties
117+
self.include = include
118+
self.exclude = exclude
113119

114120
def pre_transform(
115121
self,

test/test_automl/test_automl.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import pandas as pd
1313
import pytest
1414
import sklearn.datasets
15+
from sklearn.base import clone
1516
from smac.scenario.scenario import Scenario
1617
from smac.facade.roar_facade import ROAR
1718

@@ -426,6 +427,16 @@ def test_do_dummy_prediction(backend, dask_client, datasets):
426427
'predictions_ensemble_1_1_0.0.npy')
427428
)
428429

430+
model_path = os.path.join(backend.temporary_directory, '.auto-sklearn',
431+
'runs', '1_1_0.0',
432+
'1.1.0.0.model')
433+
434+
# Make sure the dummy model complies with scikit learn
435+
# get/set params
436+
assert os.path.exists(model_path)
437+
with open(model_path, 'rb') as model_handler:
438+
clone(pickle.load(model_handler))
439+
429440
auto._clean_logger()
430441

431442
del auto

0 commit comments

Comments
 (0)