From 50c68f6cab17747556ef2ad361658a06c9874b45 Mon Sep 17 00:00:00 2001 From: Cai Shaofeng Date: Thu, 26 Dec 2024 00:26:11 +0800 Subject: [PATCH] Add the diabetic readmission prediction application --- .../Diabetic_Readmission_Prediction/README.md | 45 +++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 examples/healthcare/application/Diabetic_Readmission_Prediction/README.md diff --git a/examples/healthcare/application/Diabetic_Readmission_Prediction/README.md b/examples/healthcare/application/Diabetic_Readmission_Prediction/README.md new file mode 100644 index 000000000..c58e6375a --- /dev/null +++ b/examples/healthcare/application/Diabetic_Readmission_Prediction/README.md @@ -0,0 +1,45 @@ + + +# Singa for Diabetic Readmission Prediction task + +## Diabetic Readmission + +Diabetic readmission is a significant concern in healthcare, with a substantial number of patients being readmitted to the hospital within a short period after discharge. This not only leads to increased healthcare costs but also poses a risk to patient well-being. + +Although diabetes is a manageable condition, early identification of patients at high risk of readmission remains a challenge. A reliable and efficient predictive model can help identify these patients, enabling healthcare providers to intervene early and prevent unnecessary readmissions. + +To address this issue, we use Singa to implement a machine learning model for predicting diabetic readmission. The dataset is from [BMC Medical Informatics and Decision-Making](https://bmcmedinformdecismak.biomedcentral.com/articles/10.1186/s12911-021-01423-y). Please download the dataset before running the scripts. + + +## Structure + +* `data` includes the scripts for preprocessing Diabetic Readmission datasets. + +* `model` includes the MLP model construction codes by creating + a subclass of `Module` to wrap the neural network operations + of each model. + +* `train_mlp.py` is the training script, which controls the training flow by + doing BackPropagation and SGD update. + +## Command +```bash +python train.py mlp diabetic +```