Commit 56db791e authored by Joaquin Torres's avatar Joaquin Torres

Realized its better to tune based on f1 to achieve good recall

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