Skip to content

Commit 9b652d5

Browse files
committed
Add solver hyperparameter in MLP classifier example, increase runtime of classifier in example_extend_preprocessing.py
1 parent c8368f5 commit 9b652d5

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

examples/example_extending_classification.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,14 @@ def __init__(self,
2727
num_nodes_per_layer,
2828
activation,
2929
alpha,
30+
solver,
3031
random_state=None,
3132
):
3233
self.hidden_layer_depth = hidden_layer_depth
3334
self.num_nodes_per_layer = num_nodes_per_layer
3435
self.activation = activation
3536
self.alpha = alpha
37+
self.solver = solver
3638
self.random_state = random_state
3739

3840
def fit(self, X, y):
@@ -45,8 +47,9 @@ def fit(self, X, y):
4547
for i in range(self.hidden_layer_depth))
4648

4749
self.estimator = MLPClassifier(hidden_layer_sizes=hidden_layer_sizes,
48-
alpha=self.alpha,
4950
activation=self.activation,
51+
alpha=self.alpha,
52+
solver=self.solver,
5053
random_state=self.random_state,
5154
)
5255
self.estimator.fit(X, y)
@@ -92,10 +95,14 @@ def get_hyperparameter_search_space(dataset_properties=None):
9295
alpha = UniformFloatHyperparameter(
9396
name="alpha", lower=0.0001, upper=1.0, default_value=0.0001
9497
)
98+
solver = CategoricalHyperparameter(
99+
name="solver", choices=['lbfgs', 'sgd', 'adam'], default_value='adam'
100+
)
95101
cs.add_hyperparameters([hidden_layer_depth,
96102
num_nodes_per_layer,
97103
activation,
98104
alpha,
105+
solver,
99106
])
100107
return cs
101108

examples/example_extending_preprocessor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def get_hyperparameter_search_space(dataset_properties=None):
9999

100100
# Fit the model using LDA as preprocessor.
101101
clf = autosklearn.classification.AutoSklearnClassifier(
102-
time_left_for_this_task=20,
102+
time_left_for_this_task=30,
103103
include_preprocessors=['LDA'],
104104
)
105105
clf.fit(X_train, y_train)

0 commit comments

Comments
 (0)