diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py index 89d82bf2f7f7a8357a858c0fd04dc51ddf2d49d5..bfd814119c92fb20bc7517359ce33d89eb7bd20f 100644 --- a/model_selection/hyperparam_tuning.py +++ b/model_selection/hyperparam_tuning.py @@ -156,8 +156,12 @@ if __name__ == "__main__": params = hyperparameters[model_name] search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='precision') search.fit(X,y) + # Access the results + results = search.cv_results_ hyperparam_df.at[model_name,'Parameters']=search.best_params_ - hyperparam_df.at[model_name,'Score']=round(search.best_score_,4) + hyperparam_df.at[model_name,'Best Precision']=round(search.best_score_,4) + hyperparam_df.at[model_name,'Mean Precision']= np.mean(results['mean_test_score']) + hyperparam_df.at[model_name,'SD']= np.std(results['std_test_score']) # Store the DataFrame in the dictionary with a unique key for each sheet sheet_name = f"{group}_{method_names[j]}" diff --git a/model_selection/output_hyperparam/hyperparamers.xlsx b/model_selection/output_hyperparam/hyperparamers.xlsx index 58c51e6a85a44afd2f2624f898c16356d7f35b4e..11584750562141b2d42534cde438f9f0cb40a299 100644 Binary files a/model_selection/output_hyperparam/hyperparamers.xlsx and b/model_selection/output_hyperparam/hyperparamers.xlsx differ