def testAddRemOptParameters(self): model = h2ohyperopt.ModelOptimizer(metric='auc') model.select_optimization_parameters(params="Default") # Adding parameters with search spaces model.add_optimization_parameters({'col_sample_rate': ('uniform', (0.5, 0.8)), 'ntrees': ('choice', [10, 20, 30])}) assert set(model.model_params.keys()) == set(['metric', 'ntrees', 'col_sample_rate']) # Removing a certain parameter model.rem_optimization_parameters('col_sample_rate') assert set(model.model_params.keys()) == set(['metric', 'ntrees'])
def testDefParameters(self): model = h2ohyperopt.ModelOptimizer(metric='auc') model.default_parameters_opt(return_dict=False) params_list = model.default_parameters_opt(return_dict=True) assert params_list == model.def_params
def testSelOptParameters(self): model = h2ohyperopt.ModelOptimizer(metric='auc') model.select_optimization_parameters(params= {'ntrees': ('choice', [10, 20, 30])}) assert set(model.model_params.keys()) == set(['metric', 'ntrees'])
def testModelParameters(self): model = h2ohyperopt.ModelOptimizer(metric='auc') model.select_optimization_parameters(params="Default") model.model_parameters_opt(return_dict=False) params_list = model.model_parameters_opt(return_dict=True) assert params_list == model.model_params
def testCheckMetricSet(self): model = h2ohyperopt.ModelOptimizer(metric='auc') assert model.check_if_metric_set() is True
def testSetMetric(self): model = h2ohyperopt.ModelOptimizer() assert model.check_if_metric_set() is False model.set_metric('auc') assert model.check_if_metric_set() is True