Commit 720f00f9 authored by Joaquin Torres's avatar Joaquin Torres

Update README.md

parent 1a32e877
# Title # Title
## Introduction ## Introduction
...
## Dealing with Class Imbalance ## Dealing with Class Imbalance
One of the primary challenges we encountered was a significant class imbalance, with a higher number of patients withdrawing from treatment compared to those staying. One of the primary challenges we encountered was a significant class imbalance, with a higher number of patients withdrawing from treatment compared to those staying.
...@@ -26,6 +26,10 @@ This repository is organized as follows: ...@@ -26,6 +26,10 @@ 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_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. * [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. * [test_models.py](./model_selection/test_models.py): Testing tuned models with test dataset.
* [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.
* [explainability](./explainability): * [explainability](./explainability):
* [fit_final_models.py](./explainability/fit_final_models.py): Saving fitted model for each selected final model. * [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_vals.py](./explainability/compute_shap_vals.py): Computing SHAP values for final models.
...@@ -34,7 +38,7 @@ This repository is organized as follows: ...@@ -34,7 +38,7 @@ This repository is organized as follows:
* [output](./explainability/output): SHAP and SHAP interaction summary plots. * [output](./explainability/output): SHAP and SHAP interaction summary plots.
## Data ## Data
The dataset has not been provided since the authors do not have permission for its sharing from the data providers. The dataset has not been provided since the authors do not have permission for its sharing from the data providers.
For any inquiry you can contact: For any inquiry you can contact:
-
-
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment