tuning pre CS

model_selection/hyperparam_tuning.py

@@ -148,11 +148,12 @@ if __name__ == "__main__":
             X = data_dic['X_train_' + method + group]
             y = data_dic['y_train_' + method + group]
             # Use group of models with class weight if needed
-            models = models_CS if j == 2 else models_simple 
+            # models = models_CS if j == 2 else models_simple 
+            models = models_CS
             # Save results: params and best score for each of the mdodels of this method and group
             hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Parameters','Score'])
             for model_name, model in models.items():
-                print(f"{group}-{method_names[j]}-{model_name}")
+                print(f"{group}-{method_names[1]}-{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')
@@ -165,9 +166,11 @@ if __name__ == "__main__":
             sheets_dict[sheet_name] = hyperparam_df
 
     # Write results to Excel file
-    with pd.ExcelWriter('./output/hyperparam_pre_ORIG.xlsx') as writer:
+    with pd.ExcelWriter('./output/hyperparam_pre_CS.xlsx') as writer:
         for sheet_name, data in sheets_dict.items():
             data.to_excel(writer, sheet_name=sheet_name)
+
+    print("Successful tuning")
     # --------------------------------------------------------------------------------------------------------