code base for the paper: Automated machine learning and explainable AI (AutoML-XAI) for metabolomics
- Clone the repository.
- Build the docker.
docker build -t Dockerfile.
- Run the docker container.
- Run jupyter notebook.
Note that the autoML models are saved asaskl_17_OC_3600.pkl
andaskl_RCC_600.pkl
for the OC and RCC dataset respectively. You will need to unzip.
Name | Description |
---|---|
automl-SHAP-RCC.ipynb | Implementation of autosklearn and SHAP for the RCC Dataset. |
RCC-model-error-analysis.ipynb | Error analysis of the autoML RCC diagnostic model. |
data-preparation.ipynb | Data preparation for the RCC metabolomics dataset. |
automl-SHAP-OC.ipynb | Implementation of autosklearn and SHAP for the OC Dataset. |
OC-model-error-analysis.ipynb | Error analysis of the autoML OC diagnostic model. |