diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py index 1016803b27d6596ca562b9419f29afe24eaf7b32..980b3114888cdeb507ec7007fff22d9274a0de22 100644 --- a/model_selection/hyperparam_tuning.py +++ b/model_selection/hyperparam_tuning.py @@ -155,7 +155,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=8, scoring='precision') + search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='precision') search.fit(X,y) hyperparam_df.at[model_name,'Parameters']=search.best_params_ hyperparam_df.at[model_name,'Score']=round(search.best_score_,4) diff --git a/model_selection/output/hyperparamers.xlsx b/model_selection/output/hyperparamers.xlsx new file mode 100644 index 0000000000000000000000000000000000000000..851287cd43a12de53eec553a8259caba36b2e19b Binary files /dev/null and b/model_selection/output/hyperparamers.xlsx differ