Commit 9d9c5650 authored by Joaquin Torres's avatar Joaquin Torres

Redone tuning with f1

parent 56db791e
......@@ -149,7 +149,7 @@ if __name__ == "__main__":
print(f"{group}-{method_names[j]}-{model_name}")
# Find optimal hyperparams for curr model
params = hyperparameters[model_name]
search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='f1')
search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=15, scoring='f1')
search.fit(X,y)
# Keep optimal parameters
best_params = search.best_params_
......
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