From 8e04e3e4b069f768571e81b5dbafe3d46c2069b1 Mon Sep 17 00:00:00 2001 From: Joaquin Torres Date: Wed, 10 Jul 2024 10:26:19 +0000 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6c76b21..ee11f5c 100644 --- a/README.md +++ b/README.md @@ -38,12 +38,13 @@ This repository is organized as follows: * [cv_metric_gen.py](./model_selection/cv_metric_gen.py): Generating cross-validation metrics and plots for each of the tuned models. * [cv_metrics_distr.py](./model_selection/cv_metrics_distr.py): Generating boxplots for each cross-validation metric and tuned model. * [test_models.py](./model_selection/test_models.py): Testing tuned models with test dataset. + * [fit_final_models.py](./model_selection/fit_final_models.py): Saving fitted model for each selected final model. * [output](./model_selection/output): * [hyperparam](./model_selection/output/hyperparam): Excel file containing the optimal hyperparameters for each model in each pipeline. * [cv_metrics](./model_selection/output/cv_metrics): Material related to the results of cross-validation: scores, ROC and Precision-Recall curves and boxplots for each metric. * [testing](./model_selection/output/testing): Material related to the results of testing the tuned models: scores, ROC and Precision-Recall curves and confusion matrices. + * [fitted_models](./model_selection/output/fitted_models): Final selected trained models. * [explainability](./explainability): - * [fit_final_models.py](./explainability/fit_final_models.py): Saving fitted model for each selected final model. * [compute_shap_vals.py](./explainability/compute_shap_vals.py): Computing SHAP values for final models. * [compute_shap_inter_vals.py](./explainability/compute_shap_inter_vals.py): Computing SHAP interaction values for final models. * [shap_plots.py](./explainability/shap_plots.py): Generating SHAP summary plots for the SHAP and SHAP interaction values computed. Comparing major differences between pre- and post-pandemic groups. -- 2.24.1