Commit 24ce5ba9 authored by Joaquin Torres's avatar Joaquin Torres

Regen hyperpameters with XGB relaxation

parent 3875a58e
......@@ -147,8 +147,6 @@ if __name__ == "__main__":
hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Best Parameters'])
for model_name, model in models.items():
print(f"{group}-{method_names[j]}-{model_name}")
if model_name != 'XGB':
continue
# Find optimal hyperparams for curr model
params = hyperparameters[model_name]
search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='precision')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment