diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py index 3ebeeaa3301d10e881121f9fd9ded82adec1c8c3..d98dfb0b4cfb2b16ead80b219334d2e4e608201a 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='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_ diff --git a/model_selection/output_hyperparam/hyperparamers.xlsx b/model_selection/output_hyperparam/hyperparamers.xlsx index 42016d3cea2f580a75dba07f0fff08367558f5dc..928a1299d2ff1c4feea5113e8ab992a74efbd959 100644 Binary files a/model_selection/output_hyperparam/hyperparamers.xlsx and b/model_selection/output_hyperparam/hyperparamers.xlsx differ