From af40f8b596deea683958c5190173d079db8b9306 Mon Sep 17 00:00:00 2001 From: joaquintb Date: Wed, 10 Jul 2024 12:22:43 +0200 Subject: [PATCH] fit final models moved to model building --- explainability/compute_shap_inter_vals.py | 2 +- explainability/compute_shap_vals.py | 2 +- .../fit_final_models.py | 0 .../output/fitted_models/post_ORIG_CW_RF.pkl | Bin .../output/fitted_models/post_ORIG_XGB.pkl | Bin .../output/fitted_models/post_OVER_XGB.pkl | Bin .../output/fitted_models/post_UNDER_XGB.pkl | Bin .../output/fitted_models/pre_ORIG_CW_RF.pkl | Bin .../output/fitted_models/pre_ORIG_XGB.pkl | Bin .../output/fitted_models/pre_OVER_XGB.pkl | Bin .../output/fitted_models/pre_UNDER_XGB.pkl | Bin 11 files changed, 2 insertions(+), 2 deletions(-) rename {explainability => model_selection}/fit_final_models.py (100%) rename {explainability => model_selection}/output/fitted_models/post_ORIG_CW_RF.pkl (100%) rename {explainability => model_selection}/output/fitted_models/post_ORIG_XGB.pkl (100%) rename {explainability => model_selection}/output/fitted_models/post_OVER_XGB.pkl (100%) rename {explainability => model_selection}/output/fitted_models/post_UNDER_XGB.pkl (100%) rename {explainability => model_selection}/output/fitted_models/pre_ORIG_CW_RF.pkl (100%) rename {explainability => model_selection}/output/fitted_models/pre_ORIG_XGB.pkl (100%) rename {explainability => model_selection}/output/fitted_models/pre_OVER_XGB.pkl (100%) rename {explainability => model_selection}/output/fitted_models/pre_UNDER_XGB.pkl (100%) diff --git a/explainability/compute_shap_inter_vals.py b/explainability/compute_shap_inter_vals.py index cebd243..fdb8fe8 100644 --- a/explainability/compute_shap_inter_vals.py +++ b/explainability/compute_shap_inter_vals.py @@ -70,7 +70,7 @@ if __name__ == "__main__": print(f"{group}-{method_names[j]}") method_name = method_names[j] model_name = model_choices[method_name] - model_path = f"./output/fitted_models/{group}_{method_names[j]}_{model_name}.pkl" + model_path = f"../model_selection/output/fitted_models/{group}_{method_names[j]}_{model_name}.pkl" # Load the fitted model from disk with open(model_path, 'rb') as file: fitted_model = pickle.load(file) diff --git a/explainability/compute_shap_vals.py b/explainability/compute_shap_vals.py index bad3860..b9625fb 100644 --- a/explainability/compute_shap_vals.py +++ b/explainability/compute_shap_vals.py @@ -70,7 +70,7 @@ if __name__ == "__main__": print(f"{group}-{method_names[j]}") method_name = method_names[j] model_name = model_choices[method_name] - model_path = f"./output/fitted_models/{group}_{method_names[j]}_{model_name}.pkl" + model_path = f"../model_selection/output/fitted_models/{group}_{method_names[j]}_{model_name}.pkl" # Load the fitted model from disk with open(model_path, 'rb') as file: fitted_model = pickle.load(file) diff --git a/explainability/fit_final_models.py b/model_selection/fit_final_models.py similarity index 100% rename from explainability/fit_final_models.py rename to model_selection/fit_final_models.py diff --git a/explainability/output/fitted_models/post_ORIG_CW_RF.pkl b/model_selection/output/fitted_models/post_ORIG_CW_RF.pkl similarity index 100% rename from explainability/output/fitted_models/post_ORIG_CW_RF.pkl rename to model_selection/output/fitted_models/post_ORIG_CW_RF.pkl diff --git a/explainability/output/fitted_models/post_ORIG_XGB.pkl b/model_selection/output/fitted_models/post_ORIG_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/post_ORIG_XGB.pkl rename to model_selection/output/fitted_models/post_ORIG_XGB.pkl diff --git a/explainability/output/fitted_models/post_OVER_XGB.pkl b/model_selection/output/fitted_models/post_OVER_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/post_OVER_XGB.pkl rename to model_selection/output/fitted_models/post_OVER_XGB.pkl diff --git a/explainability/output/fitted_models/post_UNDER_XGB.pkl b/model_selection/output/fitted_models/post_UNDER_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/post_UNDER_XGB.pkl rename to model_selection/output/fitted_models/post_UNDER_XGB.pkl diff --git a/explainability/output/fitted_models/pre_ORIG_CW_RF.pkl b/model_selection/output/fitted_models/pre_ORIG_CW_RF.pkl similarity index 100% rename from explainability/output/fitted_models/pre_ORIG_CW_RF.pkl rename to model_selection/output/fitted_models/pre_ORIG_CW_RF.pkl diff --git a/explainability/output/fitted_models/pre_ORIG_XGB.pkl b/model_selection/output/fitted_models/pre_ORIG_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/pre_ORIG_XGB.pkl rename to model_selection/output/fitted_models/pre_ORIG_XGB.pkl diff --git a/explainability/output/fitted_models/pre_OVER_XGB.pkl b/model_selection/output/fitted_models/pre_OVER_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/pre_OVER_XGB.pkl rename to model_selection/output/fitted_models/pre_OVER_XGB.pkl diff --git a/explainability/output/fitted_models/pre_UNDER_XGB.pkl b/model_selection/output/fitted_models/pre_UNDER_XGB.pkl similarity index 100% rename from explainability/output/fitted_models/pre_UNDER_XGB.pkl rename to model_selection/output/fitted_models/pre_UNDER_XGB.pkl -- 2.24.1