diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py index 4f7734de5bdb3d955d3695293852b06423615934..907d6f71035ea310fb4cc98ee7a6986acc3654b5 100644 --- a/model_selection/hyperparam_tuning.py +++ b/model_selection/hyperparam_tuning.py @@ -143,7 +143,7 @@ if __name__ == "__main__": # Use group of models with class weight if needed models = models_CS if j == 1 else models_simple # Save optimal hyperparameters for each of the models -> metrics will be computed in a different file - hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Model Name', 'Best Parameters']) + hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Best Parameters']) for model_name, model in models.items(): print(f"{group}-{method_names[j]}-{model_name}") # Find optimal hyperparams for curr model diff --git a/model_selection/output_hyperparam/hyperparamers.xlsx b/model_selection/output_hyperparam/hyperparamers.xlsx index 96eff997c434ccdd40d61a42fe25cf01d2f6fdd3..f46a8445336cd27f6baef59e7c86c2d2692fb559 100644 Binary files a/model_selection/output_hyperparam/hyperparamers.xlsx and b/model_selection/output_hyperparam/hyperparamers.xlsx differ