Skip to content

Commit 1b2dfd4

Browse files
committed
follow sklearn convention for dummy models
1 parent 7364ff3 commit 1b2dfd4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

autosklearn/evaluation/abstract_evaluator.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ def __init__(self, configuration, random_state, init_params=None):
3535
super(MyDummyClassifier, self).__init__(strategy="uniform")
3636
else:
3737
super(MyDummyClassifier, self).__init__(strategy="most_frequent")
38+
self.random_state = random_state
39+
self.init_params = init_params
3840

3941
def pre_transform(self, X, y, fit_params=None): # pylint: disable=R0201
4042
if fit_params is None:
@@ -69,6 +71,8 @@ def __init__(self, configuration, random_state, init_params=None):
6971
super(MyDummyRegressor, self).__init__(strategy='mean')
7072
else:
7173
super(MyDummyRegressor, self).__init__(strategy='median')
74+
self.random_state = random_state
75+
self.init_params = init_params
7276

7377
def pre_transform(self, X, y, fit_params=None):
7478
if fit_params is None:

0 commit comments

Comments
 (0)