def _validate_parameters(self):
"""Ensure provided input parameters are properly formatted"""
#################### target_metric ####################
self.target_metric = get_formatted_target_metric(self.target_metric, self.metrics_map)
#################### feature_selector ####################
if self.feature_selector is None:
restricted_cols = [_ for _ in [self.target_column, self.id_column] if _ is not None]
self.feature_selector = [_ for _ in self.train_dataset.columns.values if _ not in restricted_cols]
G.debug('Experiment parameters have been validated')
def _validate_parameters(self):
"""Ensure provided input parameters are properly formatted"""
self.target_metric = get_formatted_target_metric(self.target_metric,
G.Env.metrics_map,
default_dataset='oof')
def test_get_formatted_target_metric_value_error(target_metric):
with pytest.raises(ValueError):
get_formatted_target_metric(target_metric, format_metrics_map(["roc_auc_score"]))
def test_get_formatted_target_metric_type_error(target_metric):
with pytest.raises(TypeError):
get_formatted_target_metric(target_metric,
format_metrics(["roc_auc_score"]))