diff --git a/README.md b/README.md index 61bf51fdd905aa486be7440f5ae64662ef06b7dd..6e8c8486e13b367c90143b57ff80a5aa87474c70 100644 --- a/README.md +++ b/README.md @@ -6,6 +6,9 @@ This repository is organized as follows: * [EDA](./EDA): Exploring data, handling missing values, analyzing variable distribution, filtering data, feature selection, encoding variables, and performing correlation analysis. * [gen_train_data](./gen_train_data): Generating training and testing datasets [wait for final approach TBC]. * [model_selection](./model_selection): Tuning models, generating metrics from cross-validation and testing, and selecting final models. + * [hyperparam_tuning.py](./model_selection/hyperparam_tuning.py): Tunes models through a random search of hyperparameters. + * [cv_metric_gen.py](./model_selection/cv_metric_gen.py): Generates cross-validation metrics and plots for the tuned models. + * [cv_metrics_distr.py](./model_selection/cv_metrics_distr.py): * [explainability](./explicability): SHAP explainability analysis, comparing pre- and post-pandemic groups. [Outro]