README.md 1.09 KB
Newer Older
Joaquin Torres's avatar
Joaquin Torres committed
1 2 3 4 5
# Title

[Intro]

This repository is organized as follows:
Joaquin Torres's avatar
Joaquin Torres committed
6 7 8
* [EDA](./EDA): 
    * [output](./EDA/output): Plots about feature distributions, correlations and importance.
    * [EDA.ipynb](./EDA/EDA.ipynb): Notebook used for exploring and filtering data, handling missing values, encoding variables, building the final pre- and post- pandemic datasets, and generating plots for feature distributions, correlations and importance.
Joaquin Torres's avatar
Joaquin Torres committed
9 10
* [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.
Joaquin Torres's avatar
Joaquin Torres committed
11 12 13
    * [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): 
Joaquin Torres's avatar
Joaquin Torres committed
14
* [explainability](./explicability): SHAP explainability analysis, comparing pre- and post-pandemic groups.
Joaquin Torres's avatar
Joaquin Torres committed
15 16

[Outro]