This warehouse stores the main process and key code of experiments in the paper. The overall process of the paper is as follows:
- Medical image segmentation based on nnU-Net model
- Medical image feature extraction based on Pyradiomics
- Construction of clinical feature set, imaging feature set and mixed feature set
- Feature screening process
- Machine learning modeling
- Model interpretability analysis based on SHAP method
Each process is described in detail below
nnU-Net is a fully automatic segmentation framework that can complete image segmentation without special processing by the user. For details, please refer to the warehouse: https://github.com/MIC-DKFZ/nnUNet It describes in detail the various storage locations of data set files
Feature extraction using Pyradiomics requires downloading the toolkit from a third-party library https://github.com/AIM-Harvard/pyradiomics
After the download is complete, you can refer to the official sample code, or use ours(pyradiomics/extract_feature.py). You need to replace the file address in the code with your own.
After extracting the features, clinical features and image features can be combined
Feature filtering was performed using three methods, see the document for details:feature-screening/
Model building and interpretability analysis can refer to the code:model/notebook3c141729cf.ipynb