def optimize_uncertainty(trial, dataset_id):
dataset_id = str(dataset_id)
try:
gen = SpaceGenerator()
space = gen.generate_params()
space.sample_parameters(trial)
trial.set_user_attr('space', copy.deepcopy(space))
search_time, evaluation_time, memory_limit, privacy_limit, training_time_limit, inference_time_limit, pipeline_size_limit, cv, number_of_cvs, hold_out_fraction, sample_fraction, _ = generate_parameters(trial, total_search_time, my_openml_datasets)
my_random_seed = int(time.time())
X_train, X_test, y_train, y_test, categorical_indicator, attribute_names = get_data(dataset_id,
randomstate=my_random_seed)
trial.set_user_attr('data_random_seed', my_random_seed)
trial.set_user_attr('dataset_id', dataset_id)
#add metafeatures of data
my_list_constraints_values = [search_time,
evaluation_time,
memory_limit,
cv,
number_of_cvs,
ifNull(privacy_limit, constant_value=1000),
ifNull(hold_out_fraction),
sample_fraction,
training_time_limit,
inference_time_limit,
pipeline_size_limit]
metafeature_values = data2features(X_train, y_train, categorical_indicator)
features = space2features(space, my_list_constraints_values, metafeature_values)
features = FeatureTransformations().fit(features).transform(features, feature_names=feature_names)
trial.set_user_attr('features', features)
model = mp_glob.ml_model
trial.set_user_attr('predicted_target', model.predict(features))
predictions = []
for tree in range(model.n_estimators):
predictions.append(predict_range(model.estimators_[tree], features))
stddev_pred = np.std(np.matrix(predictions).transpose(), axis=1)
return stddev_pred[0]
except Exception as e:
print(str(e) + 'except dataset _ uncertainty: ' + str(dataset_id) + '\n\n')
return -np.inf
def run_AutoML(trial, X_train=None, X_test=None, y_train=None, y_test=None, categorical_indicator=None):
space = None
search_time = None
if not 'space' in trial.user_attrs:
# which hyperparameters to use
gen = SpaceGenerator()
space = gen.generate_params()
space.sample_parameters(trial)
trial.set_user_attr('space', copy.deepcopy(space))
search_time, evaluation_time, memory_limit, privacy_limit, training_time_limit, inference_time_limit, pipeline_size_limit, cv, number_of_cvs, hold_out_fraction, sample_fraction, dataset_id = generate_parameters(trial, total_search_time, my_openml_datasets)
else:
space = trial.user_attrs['space']
print(trial.params)
#make this a hyperparameter
search_time = trial.params['global_search_time_constraint']
evaluation_time = search_time
if 'global_evaluation_time_constraint' in trial.params:
evaluation_time = trial.params['global_evaluation_time_constraint']
memory_limit = 10
if 'global_memory_constraint' in trial.params:
memory_limit = trial.params['global_memory_constraint']
privacy_limit = None
if 'privacy_constraint' in trial.params:
privacy_limit = trial.params['privacy_constraint']
training_time_limit = search_time
if 'training_time_constraint' in trial.params:
training_time_limit = trial.params['training_time_constraint']
inference_time_limit = 60
if 'inference_time_constraint' in trial.params:
inference_time_limit = trial.params['inference_time_constraint']
pipeline_size_limit = 350000000
if 'pipeline_size_constraint' in trial.params:
pipeline_size_limit = trial.params['pipeline_size_constraint']
cv = 1
number_of_cvs = 1
hold_out_fraction = None
if 'global_cv' in trial.params:
cv = trial.params['global_cv']
if 'global_number_cv' in trial.params:
number_of_cvs = trial.params['global_number_cv']
else:
hold_out_fraction = trial.params['hold_out_fraction']
sample_fraction = 1.0
if 'sample_fraction' in trial.params:
sample_fraction = trial.params['sample_fraction']
if 'dataset_id' in trial.params:
dataset_id = trial.params['dataset_id']
else:
dataset_id = trial.user_attrs['dataset_id']
for pre, _, node in RenderTree(space.parameter_tree):
if node.status == True:
print("%s%s" % (pre, node.name))
if type(X_train) == type(None):
my_random_seed = int(time.time())
if 'data_random_seed' in trial.user_attrs:
my_random_seed = trial.user_attrs['data_random_seed']
X_train, X_test, y_train, y_test, categorical_indicator, attribute_names = get_data(dataset_id, randomstate=my_random_seed)
if not isinstance(trial, FrozenTrial):
my_list_constraints_values = [search_time,
evaluation_time,
memory_limit, cv,
number_of_cvs,
ifNull(privacy_limit, constant_value=1000),
ifNull(hold_out_fraction),
sample_fraction,
training_time_limit,
inference_time_limit,
pipeline_size_limit]
metafeature_values = data2features(X_train, y_train, categorical_indicator)
features = space2features(space, my_list_constraints_values, metafeature_values)
features = FeatureTransformations().fit(features).transform(features, feature_names=feature_names)
trial.set_user_attr('features', features)
dynamic_params = []
for random_i in range(5): #5
search = MyAutoML(cv=cv,
number_of_cvs=number_of_cvs,
n_jobs=1,
evaluation_budget=evaluation_time,
time_search_budget=search_time,
space=space,
main_memory_budget_gb=memory_limit,
differential_privacy_epsilon=privacy_limit,
hold_out_fraction=hold_out_fraction,
sample_fraction=sample_fraction,
training_time_limit=training_time_limit,
inference_time_limit=inference_time_limit,
pipeline_size_limit=pipeline_size_limit)
test_score = 0.0
try:
search.fit(X_train, y_train, categorical_indicator=categorical_indicator, scorer=my_scorer)
best_pipeline = search.get_best_pipeline()
if type(best_pipeline) != type(None):
test_score = my_scorer(search.get_best_pipeline(), X_test, y_test)
except:
pass
dynamic_params.append(test_score)
count_success = 0
for i_run in range(len(dynamic_params)):
if dynamic_params[i_run] > 0.0:
count_success += 1
success_rate = float(count_success) / float(len(dynamic_params))
return success_rate, search
def run_AutoML(trial, X_train=None, X_test=None, y_train=None, y_test=None, categorical_indicator=None):
space = None
search_time = None
if not 'space' in trial.user_attrs:
# which hyperparameters to use
gen = SpaceGenerator()
space = gen.generate_params()
space.sample_parameters(trial)
trial.set_user_attr('space', copy.deepcopy(space))
search_time, evaluation_time, memory_limit, privacy_limit, cv, number_of_cvs, hold_out_fraction, sample_fraction, dataset_id = generate_parameters(trial)
else:
space = trial.user_attrs['space']
print(trial.params)
#make this a hyperparameter
search_time = total_search_time
evaluation_time = search_time
memory_limit = 4
privacy_limit = None
cv = 1
number_of_cvs = 1
hold_out_fraction = None
if 'global_cv' in trial.params:
cv = trial.params['global_cv']
if 'global_number_cv' in trial.params:
number_of_cvs = trial.params['global_number_cv']
else:
hold_out_fraction = trial.params['hold_out_fraction']
sample_fraction = 1.0
if 'sample_fraction' in trial.params:
sample_fraction = trial.params['sample_fraction']
if 'dataset_id' in trial.params:
dataset_id = trial.params['dataset_id'] #get same random seed
else:
dataset_id = 31
for pre, _, node in RenderTree(space.parameter_tree):
if node.status == True:
print("%s%s" % (pre, node.name))
if type(X_train) == type(None):
my_random_seed = int(time.time())
if 'data_random_seed' in trial.user_attrs:
my_random_seed = trial.user_attrs['data_random_seed']
X_train, X_test, y_train, y_test, categorical_indicator, attribute_names = get_data(dataset_id, randomstate=my_random_seed)
if not isinstance(trial, FrozenTrial):
my_list_constraints_values = [search_time,
evaluation_time,
memory_limit, cv,
number_of_cvs,
ifNull(privacy_limit, constant_value=1000),
ifNull(hold_out_fraction),
sample_fraction]
metafeature_values = data2features(X_train, y_train, categorical_indicator)
features = space2features(space, my_list_constraints_values, metafeature_values)
features = FeatureTransformations().fit(features).transform(features, feature_names=feature_names)
trial.set_user_attr('features', features)
search = MyAutoML(cv=cv,
number_of_cvs=number_of_cvs,
n_jobs=1,
evaluation_budget=evaluation_time,
time_search_budget=search_time,
space=space,
main_memory_budget_gb=memory_limit,
differential_privacy_epsilon=privacy_limit,
hold_out_fraction=hold_out_fraction,
sample_fraction=sample_fraction)
search.fit(X_train, y_train, categorical_indicator=categorical_indicator, scorer=my_scorer)
best_pipeline = search.get_best_pipeline()
test_score = 0.0
if type(best_pipeline) != type(None):
test_score = my_scorer(search.get_best_pipeline(), X_test, y_test)
return test_score, search