diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py index 0892526c0d6a8a3845c5dda366e4753f582ed0f3..3ebeeaa3301d10e881121f9fd9ded82adec1c8c3 100644 --- a/model_selection/hyperparam_tuning.py +++ b/model_selection/hyperparam_tuning.py @@ -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='precision') + search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='f1') search.fit(X,y) # Keep optimal parameters best_params = search.best_params_