@@ -196,22 +196,35 @@ def get_hyperparameter_search_space(
196
196
197
197
add_hyperparameter (cs , epsilon , UniformFloatHyperparameter )
198
198
add_hyperparameter (cs , use_stochastic_weight_averaging , CategoricalHyperparameter )
199
+ snapshot_ensemble_flag = False
200
+ if any (use_snapshot_ensemble .value_range ):
201
+ snapshot_ensemble_flag = True
202
+
199
203
use_snapshot_ensemble = get_hyperparameter (use_snapshot_ensemble , CategoricalHyperparameter )
200
- se_lastk = get_hyperparameter (se_lastk , Constant )
201
- cs .add_hyperparameters ([use_snapshot_ensemble , se_lastk ])
202
- cond = EqualsCondition (se_lastk , use_snapshot_ensemble , True )
203
- cs .add_condition (cond )
204
+ cs .add_hyperparameter (use_snapshot_ensemble )
205
+
206
+ if snapshot_ensemble_flag :
207
+ se_lastk = get_hyperparameter (se_lastk , Constant )
208
+ cs .add_hyperparameter (se_lastk )
209
+ cond = EqualsCondition (se_lastk , use_snapshot_ensemble , True )
210
+ cs .add_condition (cond )
211
+
212
+ lookahead_flag = False
213
+ if any (use_lookahead_optimizer .value_range ):
214
+ lookahead_flag = True
204
215
205
216
use_lookahead_optimizer = get_hyperparameter (use_lookahead_optimizer , CategoricalHyperparameter )
206
217
cs .add_hyperparameter (use_lookahead_optimizer )
207
- la_config_space = Lookahead .get_hyperparameter_search_space (la_steps = la_steps ,
208
- la_alpha = la_alpha )
209
- parent_hyperparameter = {'parent' : use_lookahead_optimizer , 'value' : True }
210
- cs .add_configuration_space (
211
- Lookahead .__name__ ,
212
- la_config_space ,
213
- parent_hyperparameter = parent_hyperparameter
214
- )
218
+
219
+ if lookahead_flag :
220
+ la_config_space = Lookahead .get_hyperparameter_search_space (la_steps = la_steps ,
221
+ la_alpha = la_alpha )
222
+ parent_hyperparameter = {'parent' : use_lookahead_optimizer , 'value' : True }
223
+ cs .add_configuration_space (
224
+ Lookahead .__name__ ,
225
+ la_config_space ,
226
+ parent_hyperparameter = parent_hyperparameter
227
+ )
215
228
216
229
"""
217
230
if dataset_properties is not None:
0 commit comments