Skip to content

Commit c2d0b15

Browse files
authored
Merge pull request #1249 from solopku/patch-9
Add the diabetic readmission prediction application
2 parents 30f06a2 + 50c68f6 commit c2d0b15

File tree

1 file changed

+45
-0
lines changed
  • examples/healthcare/application/Diabetic_Readmission_Prediction

1 file changed

+45
-0
lines changed
Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
<!--
2+
Licensed to the Apache Software Foundation (ASF) under one
3+
or more contributor license agreements. See the NOTICE file
4+
distributed with this work for additional information
5+
regarding copyright ownership. The ASF licenses this file
6+
to you under the Apache License, Version 2.0 (the
7+
"License"); you may not use this file except in compliance
8+
with the License. You may obtain a copy of the License at
9+
10+
http://www.apache.org/licenses/LICENSE-2.0
11+
12+
Unless required by applicable law or agreed to in writing,
13+
software distributed under the License is distributed on an
14+
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15+
KIND, either express or implied. See the License for the
16+
specific language governing permissions and limitations
17+
under the License.
18+
-->
19+
20+
# Singa for Diabetic Readmission Prediction task
21+
22+
## Diabetic Readmission
23+
24+
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.
25+
26+
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.
27+
28+
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.
29+
30+
31+
## Structure
32+
33+
* `data` includes the scripts for preprocessing Diabetic Readmission datasets.
34+
35+
* `model` includes the MLP model construction codes by creating
36+
a subclass of `Module` to wrap the neural network operations
37+
of each model.
38+
39+
* `train_mlp.py` is the training script, which controls the training flow by
40+
doing BackPropagation and SGD update.
41+
42+
## Command
43+
```bash
44+
python train.py mlp diabetic
45+
```

0 commit comments

Comments
 (0)